Pandas将groupby列表应用于包含列表的列

vx6bjr1n  于 2022-12-21  发布在  其他
关注(0)|答案(1)|浏览(137)

我有以下 Dataframe df

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

我正在尝试以下操作:

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

我得到了以下结果:

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

我需要获取以下内容:

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

fnx2tebb1#

你可以做explode

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

相关问题