Pandas 取前 n 行 - Python

Pandas 是一个强大的数据处理库，它可以轻松处理并分析大量数据。其中，一个很常见的操作就是取前 n 行的数据。下面，我们介绍几种不同的方式来实现这个操作。

使用 head 方法

使用 head 方法可以很方便地获取前 n 行数据。它的语法格式如下：

df.head(n)

其中，df 表示数据集，n 表示要获取的行数。

示例代码：

import pandas as pd

# 创建一个 DataFrame
data = {'name': ['Alice', 'Bob', 'Charlie', 'David', 'Emily'],
        'age': [24, 42, 18, 32, 14]}
df = pd.DataFrame(data)

# 取前两行
df.head(2)

输出结果：

     name  age
0   Alice   24
1     Bob   42

使用 iloc 方法

使用 iloc 方法可以根据行号获取前 n 行数据。它的语法格式如下：

df.iloc[:n]

其中，df 表示数据集，[:n] 表示要获取的前 n 行数据。

示例代码：

import pandas as pd

# 创建一个 DataFrame
data = {'name': ['Alice', 'Bob', 'Charlie', 'David', 'Emily'],
        'age': [24, 42, 18, 32, 14]}
df = pd.DataFrame(data)

# 取前两行
df.iloc[:2]

输出结果：

    name  age
0  Alice   24
1    Bob   42

使用 loc 方法

使用 loc 方法可以根据行标签获取前 n 行数据。它的语法格式如下：

df.loc[:n]

其中，df 表示数据集，[:n] 表示要获取的前 n 行数据。需要注意的是，如果行标签不是数字，则不能使用 iloc 方法。

示例代码：

import pandas as pd

# 创建一个 DataFrame
data = {'name': ['Alice', 'Bob', 'Charlie', 'David', 'Emily'],
        'age': [24, 42, 18, 32, 14]}
df = pd.DataFrame(data)
df.index = ['a', 'b', 'c', 'd', 'e']

# 取前两行
df.loc[:'b']

输出结果：

    name  age
a  Alice   24
b    Bob   42

以上就是 Pandas 取前 n 行的几种方法，根据不同的情况选用不同的方法，可以更高效地获取目标数据。