python groupby head

在Python中，可以使用groupby函数对数据进行分组，并使用head函数获取每个分组的前几行数据。下面是一个例子代码：

import pandas as pd

# 创建一个DataFrame对象
df = pd.DataFrame({'A': ['a', 'b', 'a'], 'B': ['b', 'a', 'c'], 'C': [1, 2, 3})

# 按照'A'和'B'列进行分组
grouped = df.groupby(['A', 'B'])

# 定义需要取出的行数
n = 2

# 遍历每个分组
for name, group in grouped:
    # 获取该分组的前n行数据
    sample = group.head(n)
    # 输出结果
    print(f'Group: {name}\n{sample}\n')

这段代码会根据'A'和'B'列的值对DataFrame进行分组，然后使用head函数取出每个分组的前2行数据，并输出结果。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *3* [python pandas的groupBy 使用](https://blog.csdn.net/randompeople/article/details/101924517)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
- *2* [使用Python中的head函数获取数据集中各分组的前N行样本](https://blog.csdn.net/qq_33885122/article/details/130393007)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]