创建单词列表并按索引分组

Question

我有一列索引，每个索引都有对应的单词：

等等

我想按索引将它们按如下方式分组：对于id 1-[word1，word2，word3]对于id 2-[word4，word5]

诸如此类

并提取到CSV文件

我有这个代码：

df = pd.DataFrame(data)
d={"word":"first"}
df_new = df.groupby(df['id'], as_index=False).aggregate(d).reindex(columns=df['word'])
print (df_new)

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

我需要做什么改变才能让它起作用？

提前谢谢你

Answer 1

# Import Dependencies
import pandas as pd

# Create DataFrame
data = {'id': [1, 1, 1, 2, 2], 'word': ['word1', 'word2', 'word3', 'word4', 'word5']}
df = pd.DataFrame(data)

# Groupby and Merge
df = df.groupby('id', as_index=False).agg({'word' : ','.join})  

# Result

    id  word
0   1   word1,word2,word3
1   2   word4,word5