Python 合并列-行中的值(单元格类型为列表)

Question

我正在使用 python 聚合数据作为测试。对于每个columnA 值，我希望有一行包含columnB 和columnC 中的值。在完成此操作并从 stackoverflow 获得建议后，效果很好:

df = pd.DataFrame({'columnA':[1111,1111,2222,3333,4444,4444,5555,6666],
                   'columnB':['AAAA','AAAA','BBBB','BBBB','CCCC','CCCC','BBBB','CCCC'],
                   'columnC':['one','two','one','one','one','one','two','one'],
                   'NUM1':[1,3,5,7,1,0,4,5],
                   'NUM2':[5,3,6,9,2,4,1,1],
                   'W':list('aaabbbbb')})

d = {'columnB':lambda x: x.tolist(), 'columnC':lambda x: x.tolist() }
df1 = df.groupby('columnA').agg(d)
print(df1)

如果列表显示两个相同的值(如图像中的绿色箭头所示)，我现在要做的是合并每个单元格内的值。

所以，我尝试了这个:

d = {'columnB':lambda x: set(x.tolist()), 'columnC':lambda x: x.tolist() }
df1 = df.groupby('columnA').agg(d)
print(df1)

但我不确定列值的格式。我正在考虑再次将列转换为列表:

d = {'columnB':lambda x: list(set(x.tolist())), 'columnC':lambda x: x.tolist() }
df1 = df.groupby('columnA').agg(d)
print(df1)

您认为这是一个好的做法吗？我正在尝试了解有关聚合技术的更多信息。

Answer 1

我会做什么独特

d = {'columnB':'unique', 'columnC':'unique' }
df1 = df.groupby('columnA').agg(d)
df1
Out[573]: 
        columnB     columnC
columnA                    
1111     [AAAA]  [one, two]
2222     [BBBB]       [one]
3333     [BBBB]       [one]
4444     [CCCC]       [one]
5555     [BBBB]       [two]
6666     [CCCC]       [one]