pandas.DataFrame.take

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

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

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

Параметры:

indices:array-like

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

axis:{0 или ‘index’, 1 или ‘columns’, None}, по умолчанию 0

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

**kwargs

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

Возвращает:

тип, такой же, как вызывающий объект

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

См. также

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