考虑这个输入 df
my_input_df = pd.DataFrame({
'export_services': [[1],[2,4,5],[4,6], [2,4,5],[1]],
'seaport':['china','africa','europe', 'mexico','europe'],
'price_of_fish':['100','200','250','125','75']})
如何在包含列表的列上进行分组并将其他列合并为一个列表?
my_output_df = pd.DataFrame({
'export_services': [[1],[2,4,5],[4,6]],
'seaport':[['china','europe'],['africa','mexico'],'europe'],
'price_of_fish':[['100','75'],'200',['250','125']]})
我试过
my_input_df.groupby('export_services').apply(list)
这给了
类型错误:不可哈希类型:"列表">
有什么想法吗?
注意:如果my_output_df中的所有分组行都是列表,即使对于单个条目也是如此。
首先,转换为tuple
,可以散列:
df.export_services = df.export_services.apply(tuple)
groupby
agg
df.groupby('export_services').agg(list).reset_index()
export_services seaport price_of_fish
0 (1,) [china, europe] [100, 75]
1 (2, 4, 5) [africa, mexico] [200, 125]
2 (4, 6) [europe] [250]