Python - 在数据框中查找具有相同 id 的元素并将其组合在一起

Question

我有一个这样的数据框：id  value1 value2 0     1     1 1     2     3 2     1     4 3     1     5 4     2     1我希望它是这样的：id  value1 value2  count 0     1     1,4,5   3 1     2     3,1     2

Answer 1

像这样的东西也可以工作：

In [2468]: df.value2 = df['value2'].apply(str)

In [2494]: res = df.groupby('value1')['value2'].apply(lambda x:','.join(x)).reset_index()

In [2498]: res['count'] = df.groupby('value1').size().reset_index()[0]

In [2499]: res

Out[2499]: 

   value1 value2  count

0       1  1,4,5      3

1       2    3,1      2

Answer 2

使用aggwithjoin和聚合size，但有必要将列转换为字符串：

tups = [('value2', lambda x: ','.join(x.astype(str))), ('count', 'size')]

df1 = df.groupby('value1')['value2'].agg(tups).reset_index()

print (df1)

   value1 value2  count

0       1  1,4,5      3

1       2    3,1      2

选择：

tups = [('value2', ','.join), ('count', 'size')]

df1 = df['value2'].astype(str).groupby(df['value1']).agg(tups).reset_index()