Pandas DataFrame take() 方法

定义和用法

take() 方法从 DataFrame 返回指定的行。

take() 方法需要行位置,即使行具有命名索引。

实例

从 DataFrame 返回第 5 行和第 10 行:

在此例中,我们使用名为 data.csv 的 .csv 文件

import pandas as pd

df = pd.read_csv('data.csv')

print(df.take([5, 10]))

亲自试一试

语法

dataframe.take(indices, axis, kwargs)

参数

axis 参数是一个关键字参数

参数 描述
indices 列表 必需。要返回的位置的数组。
axis
  • 0
  • 1
  • 'index'
  • 'columns'
可选。默认为 0。从中选择元素的轴。
kwargs 为了与 numpy.take() 兼容。对输出没有影响。

返回值

包含选定行/列的 DataFrame