python - Pandas :在组内聚合之前排序

Question

我有以下 Pandas 数据框:

A        B         C
A        A         Test1
A        A         Test2
A        A         XYZ
A        B         BA
A        B         AB
B        A         AA

我想对这个数据集进行两次分组:首先通过 A 和 B 连接 C 中的组，然后仅在 A 上获得仅由 A 列定义的组。结果如下所示:

A        A        Test1,Test2,XYZ
A        B        AB, BA
B        A        AA

最后的结果应该是:

A        A,A:(Test1,Test2,XYZ), A,B:(AB, BA)
B        B,A:(AA)

连接本身有效，但排序似乎无效。

谁能帮我解决这个问题？

亲切的问候。

Answer 1

使用groupby + join

s1=df.groupby(['A','B']).C.apply(','.join)

s1
Out[421]: 
A  B
A  A    Test1,Test2,XYZ
   B              BA,AB
B  A                 AA
Name: C, dtype: object

s1.reset_index().groupby('A').apply(lambda x : x.set_index(['A','B'])['C'].to_dict())
Out[420]: 
A
A    {('A', 'A'): 'Test1,Test2,XYZ', ('A', 'B'): 'B...
B                                   {('B', 'A'): 'AA'}
dtype: object