python - 分组并附加列表和字符串

Question

我正在尝试对“value_1”列中的值进行分组。但我的最后一个专栏是由列表组成的。当我尝试使用“value_1”列进行分组时，由列表组成的列消失了。

数据框:

 value_1:        value_2:           value_3:               list: 
 american     california, nyc      walmart, kmart      [supermarket, connivence] 
 canadian         toronto            dunkinDonuts      [coffee]
 american          texas                               [state]
 canadian                             walmart          [supermarket] 
   ...              ...                 ...              ....

我的预期输出是:

value_1:        value_2:              value_3:             list: 
american   california, nyc, texas   walmart, kmart      [supermarket, connivence, state] 
canadian         toronto         dunkinDonuts, walmart  [coffee, supermarket]

谢谢!

Answer 1

你可以 groupby value_1 并使用以下函数聚合包含字符串的列:

def str_cat(x):
    return x.str.cat(sep=', ')

并使用 GroupBy.sum在 list 列中附加列表:

df.replace('',None).groupby('value_1').agg({'list':'sum', 'value_2': str_cat,
                                            'value_3': str_cat})

                        list                       value_2  \
value_1                                                              
american  [supermarket, connivence, state]  california, nyc, texas   
canadian             [coffee, sipermarket]          toronto, texas   

                    value_3  
value_1                                 
american  walmart, kmart, dunkinDonuts  
canadian         dunkinDonuts, walmart