可以使用pandas中的groupby函数以及nunique函数来实现。
示例代码如下:
import pandas as pd
# 构造示例数据
data = {'column1': ['A', 'B', 'A', 'C', 'C', 'B'],
'column2': [1, 2, 1, 3, 3, 4]}
df = pd.DataFrame(data)
# 对column1列内不同对进行分组,并计算每个组的数目
result = df.groupby(['column1', 'column2']).size().reset_index(name='count')
print(result)
输出结果为:
column1 column2 count
0 A 1 2
1 B 2 1
2 B 4 1
3 C 3 2
以上代码中,groupby
函数使用列名列表['column1', 'column2']
进行分组,nunique
函数计算每个分组内不同对的数目,最终返回每个分组的计数值。