python - Pandas 数据框分组到同一行中具有分组观察的新数据框

Question

我正在 build 一个 df其中观察分组two by two如下:

group user amount
ALFA   x    2.3
ALFA   y    1.9
BETA   w    1.8
GAMMA  z    1.3
BETA   s    0.9
GAMMA  q    0.4

我想要一个新的 df与属于同一组的用户在同一行，如下图:

user1 user2 amount1 amount2
 x      y     2.3     1.9
 w      s     1.8     0.9
 z      q     1.3     0.4

值不需要以任何特定方式排序。
我试过 df.groupby('group').agg(['min','max']) ，但这会按字母顺序对用户进行排序，而不是对应的一对用户数量。
有人知道如何用 Pandas 解决这个问题吗？

Answer 1

对列 group 上的数据框进行分组然后在列表理解中使用 np.hstack水平堆叠对应的users和 amount并创建一个新的数据框:

d = pd.DataFrame([
       np.hstack([g['user'], g['amount']]) for _, g in df.groupby('group')],
       columns=['user1', 'user2', 'amount1', 'amount2'])

结果:

  user1 user2  amount1  amount2
0     x     y      2.3      1.9
1     w     s      1.8      0.9
2     z     q      1.3      0.4