python - 如何合并pandas中的重复行

Question

这个问题已经有答案了:

Concatenate strings from several rows using Pandas groupby (8 个回答)

How to group dataframe rows into list in pandas groupby (16 个回答)

已关闭 8 个月前。

import pandas as pd
df = pd.DataFrame({'id':['A','A','A','B','B','B','C'],'name':[1,2,3,4,5,6,7]})
print(df.to_string(index=False))

到目前为止，上述代码的输出是:

id  name
 A     1
 A     2
 A     3
 B     4
 B     5
 B     6
 C     7

但我期望它的输出如下:

id    name
A     1,2,3
B     4,5,6
C     7

我不知道该怎么做，我尝试了其他几个代码，但对我不起作用。请帮忙解决这个问题。

Answer 1

如果您想要逗号分隔的值列表，可以使用 join 进行聚合，请注意，您必须首先将值转换为字符串:

df2 = df.groupby('id', as_index=False).agg(lambda x: ','.join(map(str, x)))
print(df2.to_string(index=False))

输出:

id  name
 A 1,2,3
 B 4,5,6
 C     7

如果您只想要一个值列表，请使用 list 进行聚合:

df2 = df.groupby('id', as_index=False).agg(list)
print(df2.to_string(index=False))

输出:

id      name
 A [1, 2, 3]
 B [4, 5, 6]
 C       [7]