以下是一个示例代码,演示如何根据组ID条件计算行数:
import pandas as pd
# 创建示例数据
data = {'GroupID': [1, 1, 2, 2, 2, 3, 3, 3, 3],
'Value': [10, 20, 30, 40, 50, 60, 70, 80, 90]}
df = pd.DataFrame(data)
# 按组ID条件计算行数
row_count = df.groupby('GroupID').size().reset_index(name='RowCount')
print(row_count)
输出结果为:
GroupID RowCount
0 1 2
1 2 3
2 3 4
这段代码首先创建了一个示例的DataFrame,其中包含了一个名为GroupID的列和一个名为Value的列。然后,使用groupby
方法按GroupID分组,并使用size
方法计算每个组的行数。最后,使用reset_index
方法将结果转换为一个新的DataFrame,并给该列命名为RowCount。
上一篇:按组ID简单扩展数据表的行