Метод take
Метод take применяется к DataFrame
и возвращает строки (или столбцы) на основе переданного массива
позиционных индексов. Первым параметром передаётся список позиций
(индексов, начиная с нуля), которые нужно выбрать.
Вторым параметром можно указать ось: 0 для строк
и 1 для столбцов. Если параметр оси не указан,
выбираются строки.
Синтаксис
df.take(indices[, axis=0])
Пример
Давайте выберем строки с индексами 1 и 3 из таблицы:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 30, 40, 50]
})
print(df.take([1, 3]))
Результат выполнения кода:
a b
1 2 20
3 4 40
Пример
Теперь выберем столбцы с индексами 0 и 1,
используя параметр оси:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 30, 40, 50],
'c': [100, 200, 300, 400, 500]
})
print(df.take([0, 2], axis=1))
Результат выполнения кода:
a c
0 1 100
1 2 200
2 3 300
3 4 400
4 5 500
Пример
Метод take также может принимать отрицательные индексы:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 30, 40, 50]
})
print(df.take([-1, -2]))
Результат выполнения кода:
a b
4 5 50
3 4 40
