📅  最后修改于: 2023-12-03 14:57:17.167000             🧑  作者: Mango
在数据分析和处理中,我们常常需要获取 DataFrame 中的部分数据进行调查或者处理。针对给定的 DataFrame,本文将介绍如何获取前3行数据。
我们将使用 Pandas 库来创建 DataFrame,本文给出一个示例。可以使用以下代码创建一个 DataFrame。
import pandas as pd
data = {'名字': ['Kobe', 'James', 'Curry', 'Jordan', 'Durant', 'Obama', 'Trump'],
'国家': ['美国', '美国', '美国', '美国', '美国', '美国', '美国'],
'出生年份': [1978, 1984, 1988, 1963, 1988, 1961, 1946],
'职业': ['篮球运动员', '篮球运动员', '篮球运动员', '篮球运动员', '篮球运动员', '政治家', '政治家']}
df = pd.DataFrame(data)
print(df)
输出:
名字 国家 出生年份 职业
0 Kobe 美国 1978 篮球运动员
1 James 美国 1984 篮球运动员
2 Curry 美国 1988 篮球运动员
3 Jordan 美国 1963 篮球运动员
4 Durant 美国 1988 篮球运动员
5 Obama 美国 1961 政治家
6 Trump 美国 1946 政治家
Pandas 的 DataFrame 对象具有许多有用的方法。.head() 方法用于获取 DataFrame 中的前3行数据。只需传递头几行您要检查的整数即可,例如,输入 5 即可返回 DataFrame 的前5行。
df.head(3)
输出:
名字 国家 出生年份 职业
0 Kobe 美国 1978 篮球运动员
1 James 美国 1984 篮球运动员
2 Curry 美国 1988 篮球运动员
另一种选择是使用 Pandas iloc 属性。iloc 属性返回 DataFrame 中的指定行和列。例如,一行是从 0 开始的数字,第一列是 0,以此类推。
df.iloc[:3]
输出:
名字 国家 出生年份 职业
0 Kobe 美国 1978 篮球运动员
1 James 美国 1984 篮球运动员
2 Curry 美国 1988 篮球运动员
以上就是获取给定DataFrame的前3行的方法。如果您想获取其他数量的行,无论是更少还是更多,只需相应地修改 3 为您要检查的整数即可。问题解决了,请享用!