如何合并数据框行，并将它们的字符串列合并到列表中？

Question

假设我有一个 Pandas 数据框：

index  name      A
0      one       a
1      two       a
2      one       b
3      two       a

如何合并具有相同 'name' 的行，以便新列 A 是与每个 'name' 关联的所有 A 的列表？因此，输出将是：

index  name      A
0      one       [a, b]
1      two       [a]

Answer 1

这将按名称列分组并将 a 中的所有值设置为唯一列表

import pandas as pd
import numpy as np

df.groupby(['name'])['A'].apply(lambda x : np.unique(list(x))).reset_index()

How to combine dataframe rows, and combine their string column into list?