pandas groupby 分组选前5组

在pandas中，groupby函数用于按照指定的列或多个列对数据进行分组。要选择前5组，可以使用head()函数来获取每个分组的前5行数据。

下面是一个示例代码：

import pandas as pd

# 创建一个DataFrame
data = {'Group': ['A', 'A', 'B', 'B', 'C', 'C', 'C', 'D', 'D', 'D'],
        'Value': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}
df = pd.DataFrame(data)

# 按照Group列进行分组，并选择每个分组的前5行数据
result = df.groupby('Group').head(5)

print(result)

输出结果为：

  Group  Value
0     A      1
1     A      2
2     B      3
3     B      4
4     C      5
5     C      6
6     C      7
7     D      8
8     D      9
9     D     10