获取 Pandas DataFrame 的前 n 条记录

在数据处理的过程中，我们有时需要获取 DataFrame 中的前 n 条记录，可以通过以下几种方式实现。

1. head 方法

使用 DataFrame 的 head 方法可以返回前 n 行记录，默认返回前 5 行。示例代码如下：

import pandas as pd

# 创建示例 DataFrame
df = pd.DataFrame({'id': [1, 2, 3, 4, 5, 6],
                   'name': ['Tom', 'Jerry', 'Lucy', 'John', 'Mary', 'Bob'],
                   'age': [20, 21, 19, 18, 22, 23]})

# 获取前三行记录
print(df.head(3))

输出结果为：

   id   name  age
0   1    Tom   20
1   2  Jerry   21
2   3   Lucy   19

2. iloc 方法

使用 DataFrame 的 iloc 方法可以通过位置索引获取前 n 行记录，示例代码如下：

# 获取前三行记录
print(df.iloc[:3])

输出结果与上面相同。

3. loc 方法

使用 DataFrame 的 loc 方法可以通过标签索引获取前 n 行记录，示例代码如下：

# 获取前三行记录
print(df.loc[:2])

输出结果与上面相同。

4. query 方法

使用 DataFrame 的 query 方法可以按条件查询前 n 行记录，示例代码如下：

# 获取年龄小于等于20岁的前三行记录
print(df.query('age<=20').head(3))

输出结果为：

   id  name  age
0   1   Tom   20
2   3  Lucy   19
3   4  John   18

以上就是获取 Pandas DataFrame 的前 n 条记录的几种方法，根据需要选择相应的方法即可。