使用 Python-Pandas 中的 head() 和 tail() 方法选择 Dataframe 中的第一行或最后 N 行

在Python中，Pandas是一种流行的数据分析工具。Pandas提供了DataFrame数据结构，用于处理结构化数据，类似于SQL中的表格。我们可以使用Pandas的head()和tail()方法选择DataFrame数据中的第一行或最后N行。

head() 方法

head()方法返回DataFrame数据中的前N行，默认情况下，N=5。如果我们需要返回更多的行，则在括号内指定行数，如下所示：

import pandas as pd

data = {"Name": ["John", "Mike", "Sarah", "Kate", "Lee"],
        "Age": [25, 36, 28, 30, 33],
        "Salary": [50000, 80000, 65000, 70000, 75000]}

df = pd.DataFrame(data)

print(df.head())   # 返回前5行数据
print(df.head(3))  # 返回前3行数据

输出结果

    Name  Age  Salary
0   John   25   50000
1   Mike   36   80000
2  Sarah   28   65000
3   Kate   30   70000
4    Lee   33   75000

    Name  Age  Salary
0   John   25   50000
1   Mike   36   80000
2  Sarah   28   65000

tail() 方法

tail()方法返回DataFrame数据中的后N行，默认情况下，N=5。如果我们需要返回更多的行，则在括号内指定行数，如下所示：

import pandas as pd

data = {"Name": ["John", "Mike", "Sarah", "Kate", "Lee"],
        "Age": [25, 36, 28, 30, 33],
        "Salary": [50000, 80000, 65000, 70000, 75000]}

df = pd.DataFrame(data)

print(df.tail())   # 返回后5行数据
print(df.tail(3))  # 返回后3行数据

输出结果

    Name  Age  Salary
0   John   25   50000
1   Mike   36   80000
2  Sarah   28   65000
3   Kate   30   70000
4    Lee   33   75000

    Name  Age  Salary
2  Sarah   28   65000
3   Kate   30   70000
4    Lee   33   75000

总结

使用Pandas的head()和tail()方法可以帮助我们快速查看DataFrame数据的前几行或后几行数据。这在我们需要快速了解数据结构和内容时非常有用。它还提供了一个方便的方式来检查数据是否正确载入。