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

Question

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

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

等等。

我想以这种方式按索引对它们进行分组： 对于 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