以下是一个示例代码,用于按组统计返回的记录数:
import pandas as pd
# 创建示例数据
data = {'Group': ['A', 'B', 'A', 'B', 'A', 'B'],
'Value': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)
# 按组统计记录数
count = df.groupby('Group').size()
print(count)
输出:
Group
A 3
B 3
dtype: int64
这段代码首先创建了一个包含两列的DataFrame,其中一列是组别,另一列是值。然后,使用groupby
方法按组对数据进行分组,并使用size
方法计算每个组的记录数。最后,将结果打印出来。