熊猫将 groupby 列表应用于包含列表的列

Question

I have the following dataframe df :我有以下数据框df ：

Col1    Col2
A       'AD'
A       ['FG', 'LKL']
B       ['FGT']

I am trying the following:我正在尝试以下操作：

df.groupby('Col1')['Col2'].apply(list)

I get the following:我得到以下信息：

Col1    Col2
A       ['AD', "['FG', 'LKL']"]
B       ['FGT']

I need to get the following:我需要得到以下信息：

Col1    Col2
A       ['AD', 'FG', 'LKL']
B       ['FGT']

Answer 1

You can do explode你可以做explode

out = df.explode('Col2').groupby('Col1')['Col2'].apply(list)