获取Pandas数据框架的前n条记录

  • Post category:Python

获取Pandas数据框架的前n条记录可以使用Pandas提供的head()方法。

步骤如下:

  1. 首先需要导入Pandas库
import pandas as pd
  1. 创建数据框架并加载数据

这里以加载CSV格式数据为例,假设数据存储在data.csv中。

df = pd.read_csv('data.csv')
  1. 使用head()方法获取前n条记录

head()方法的参数为n,代表获取前n条记录。例如,获取前5条记录可以使用如下代码:

n = 5
df_head = df.head(n)
print(df_head)

运行结果为:

   ID  Name  Age  Gender
0   1   Bob   25    Male
1   2  John   30    Male
2   3   Sue   28  Female
3   4  Lily   27  Female
4   5   Tom   35    Male
  1. 注意事项

需要注意的是,当 n 大于数据框架的总记录数时,head()方法会返回所有记录。

另外,需要确保导入的数据格式正确,否则会导致数据框架加载失败。