python - 使用 Pandas 列出第二列的所有值和第一列的计数

Question

鉴于此数据框:

df = pd.DataFrame([[1, 2], [1, 3], [4, 6]], columns=['A', 'B'])

我可以使用groupby来显示组合组的大小:

df.groupby(['A','B']).size()

A  B
1  2    1
   3    1
4  6    1

如何将 B 的唯一值组合到一个列表中，并像这样显示 A 的大小？

A  B
1  2,3  2
4  6    1

Answer 1

用途:

df['B'].astype(str).groupby(df['A']).agg([','.join,'size'])
Out[134]: 
  join  size
A           
1  2,3     2
4    6     1