你可以使用 pd.DataFrame.head(n)
方法来获取指定行数的 pd.DataFrame
,其中 n
是你想要获取的行数。
例如:
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
# 获取前两行
df_head_2 = df.head(2)
print(df_head_2)
输出:
A B C
0 1 4 7
1 2 5 8