集团使用多个键

Question

这由多个列是我希望我的决赛桌的键1和键2.独特的组合我dataframe-

Key1 Key2 Value1 Value2 
A Alpha 16 12345 
B Beta 12 123 
A Alpha 15 1456 
A Beta 14 12345 

我不得不俱乐部值1和值2的基础结构如下：

Key1 Key2 Value1 Value2 
A Alpha {16,15} {12345,1456} 
B Beta {12}  {123} 
A Beta {14}  {12345} 

请建议一个代码。非常感谢。编码级别 - 8天。

Answer 1

您必须编写自己的自定义聚合函数。 agg通过不是分组列的每个系列并返回单个值。这里我们使用set作为聚合器。

df.groupby(['Key1', 'Key2']).agg(lambda x: set(x.values)) 

    Key1 Key2 Value1   Value2 
0 A Alpha {16, 15} {1456, 12345} 
1 A Beta  {14}  {12345} 
2 B Beta  {12}   {123}