python - 对 A 列进行分组后，将 B 列中的唯一值转置到新列

Question

我有以下数据框:

COL_A   COL_B
1   a
1   b
2   a
2   b
3   a
3   b
4   c
4   d
5   c
5   d

我正在尝试对 COL_A 中的值进行分组，并将 COL_B 中的唯一值旋转到新列

COL_A   COL_B   COL_C
1   a   b
2   a   b
3   a   b
4   c   d
5   c   d

我已经使用 df.groupby 和 df.pivot 尝试了多种不同的技术，但似乎无法获得我想要的结果。

Answer 1

由于每个唯一的 COL_A 值总是恰好出现两次，因此您可以重新调整系列并提供给 pd.DataFrame 构造函数:

df = df.sort_values('COL_A')

arr = df['COL_B'].values.reshape(-1, 2)

res = pd.DataFrame({'COL_A': df['COL_A'].unique(),
                    'COL_B': arr[:, 0],
                    'COL_C': arr[:, 1]})

print(res)

   COL_A COL_B COL_C
0      1     a     b
1      2     a     b
2      3     a     b
3      4     c     d
4      5     c     d