数据框的第一行(通过参数指定 n) - Python

在 Python 中使用 Pandas 库操作数据时，经常需要对数据框的某些行进行操作。其中一个常见的需求是获取数据框的第一行或前几行。本文将介绍如何通过参数指定获取数据框的第一行。

DataFrame.head() 方法

Pandas 库提供了 DataFrame.head() 方法，可以用于获取数据框的前几行，默认获取的行数为 5。以下是该方法的使用方式：

import pandas as pd

# 创建数据框
data = {'Name': ['Tom', 'Jack', 'Steve', 'Ricky'],'Age': [28, 34, 29, 42]}
df = pd.DataFrame(data)

# 获取前两行
head_2 = df.head(2)
print(head_2)

输出结果为：

   Name  Age
0   Tom   28
1  Jack   34

参数指定获取首行

上述例子中，我们通过 df.head(2) 获取了前两行，如果想获取首行，只需将参数设置为 1：

import pandas as pd

# 创建数据框
data = {'Name': ['Tom', 'Jack', 'Steve', 'Ricky'],'Age': [28, 34, 29, 42]}
df = pd.DataFrame(data)

# 获取首行
head_1 = df.head(1)
print(head_1)

输出结果为：

  Name  Age
0  Tom   28

总结

在 Python 中，通过 Pandas 库的 DataFrame.head() 方法，我们可以方便地获取数据框的前几行，包括首行。只需要通过设置参数来指定需要获取的行数即可，具体的使用方法需要根据实际需求进行灵活调整。