python - pandas - 组合行字符串

Question

我有一个这样的数据框:

    id        text
    1         DM,HTN,Enlarged prostate
    2         hypertensive and on regular treatment
    2         LBP
    3         DM,HTN,Enlarged prostate

我想像这样合并相同 id 的文本:

   id        text
   1         DM,HTN,Enlarged prostate
   2         hypertensive and on regular treatment LBP
   3         DM,HTN,Enlarged prostate

id 2 的文本合并。我怎样才能做到这一点？任何帮助将不胜感激。

Answer 1

您可以使用 group by 函数来执行此操作

grouped_df = df.groupby("id").apply(lambda x: "%s" % ' '.join(x['text']))