pandas.DataFrame.take

DataFrame.take(indices, axis=0, is_copy=None, **kwargs)[source]

Возвращает элементы по заданным порядковым индексам вдоль оси.

Это означает, что мы не индексируем по фактическим значениям в атрибуте индекса объекта. Мы индексируем по фактическому положению элемента в объекте.

Параметры

indices:array-like

Массив целых чисел, указывающий позиции для взятия элементов.

axis:{0 или ‘индекс’, 1 или ‘колонки’, None}, по умолчанию 0

Ось, по которой выбираются элементы. 0 означает выбор строк, 1 означает выбор столбцов. Для Series этот параметр не используется и по умолчанию равен 0.

is_copy:bool

Перед pandas 1.0, is_copy=False можно было указать, чтобы убедиться, что возвращаемое значение является фактической копией. Начиная с pandas 1.0, take всегда возвращает копию, и ключевое слово поэтому устарело.

Устарело начиная с версии 1.0.0.

**kwargs

Для совместимости с numpy.take(). Не оказывает никакого влияния на вывод.

Возвращает

taken:тип, совпадающий с типом вызывающего объекта

Массивоподобный объект, содержащий взятые из объекта элементы.

См. также

DataFrame.loc

Выбор подмножества DataFrame по меткам.

DataFrame.iloc

Выбор подмножества DataFrame по позициям.

numpy.take

Извлечение элементов из массива вдоль оси.

Примеры

>>> df = pd.DataFrame([('falcon', 'bird', 389.0),
...                    ('parrot', 'bird', 24.0),
...                    ('lion', 'mammal', 80.5),
...                    ('monkey', 'mammal', np.nan)],
...                   columns=['name', 'class', 'max_speed'],
...                   index=[0, 2, 3, 1])
>>> df
     name   class  max_speed
0  falcon    bird      389.0
2  parrot    bird       24.0
3    lion  mammal       80.5
1  monkey  mammal        NaN

Извлечение элементов с позициями 0 и 3 вдоль оси 0 (по умолчанию).

Обратите внимание, как фактические выбранные индексы (0 и 1) не соответствуют нашим выбранным индексам 0 и 3. Это потому, что мы выбираем 0-ю и 3-ю строки, а не строки, индексы которых равны 0 и 3.

>>> df.take([0, 3])
     name   class  max_speed
0  falcon    bird      389.0
1  monkey  mammal        NaN

Извлечение элементов с индексами 1 и 2 вдоль оси 1 (выбор столбцов).

>>> df.take([1, 2], axis=1)
    class  max_speed
0    bird      389.0
2    bird       24.0
3  mammal       80.5
1  mammal        NaN

Мы можем извлекать элементы, используя отрицательные целые числа для положительных индексов, начиная с конца объекта, как и в списках Python.

>>> df.take([-1, -2])
     name   class  max_speed
1  monkey  mammal        NaN
3    lion  mammal       80.5