python - 根据多列分组聚合列的唯一值并计算唯一值 - 熊猫

Question

ID col1 col2    col3
I1 1    0       1 
I2 1    0       1 
I3 0    1       0 
I4 0    1       0 
I5 0    0       1 

这是我的数据框。我期待根据 col1、col2、col3 的 group by 聚合 ID 值，并且还想要一个计数列。

预期输出：

ID_List      Count 
[I1,I2]       2
[I3,I4]       2
[I5]          1

我的代码

cols_to_group = ['col1','col2','col3']
data = pd.DataFrame(df.groupby(cols_to_group)['id'].nunique()).reset_index(drop=True)
data.head()

   ID
0  2
1  2
2  1

标签： pythonpandaspandas-groupby

Answer 1

你可以做一个groupby.agg()：

df.groupby(['col1','col2','col3'], sort=False).ID.agg([list,'count'])

输出：

                    list  count
col1 col2 col3                 
1    0    1     [I1, I2]      2
0    1    0     [I3, I4]      2
     0    1         [I5]      1