如何修改Pandas代码实现按id分组汇总单词并导出CSV

修改方案 #

你现有代码的问题如下：

• 聚合规则使用first，仅能返回每个id分组的第一个单词，无法汇总同id下所有单词为列表
• 末尾的reindex操作逻辑错误，会导致生成的结果列完全不符合预期

修改后可实现需求的完整代码如下：

import pandas as pd

df = pd.DataFrame(data)
# 按id分组，将同id的word汇总为列表
df_new = df.groupby('id', as_index=False)['word'].agg(list)

# 若希望CSV里存储的是无括号的纯逗号分隔字符串，可替换上面的聚合代码为下面这行
# df_new = df.groupby('id', as_index=False)['word'].agg(lambda x: ','.join(x))

print(df_new)
df_new.to_csv('test.csv', sep='\t', encoding='utf-8', index=False)

内容的提问来源于stack exchange，提问作者d12