python - Pandas /Python : How to Transpose Duplicate Rows to Columns and Preserve Order?

Question

我有两列。 第一列具有值 A、B、C、D，第二列值对应于 A、B、C 和 D。

我想将 A、B、C 和 D 转换/转置为名为 A、B、C、D 的 4 列，并具有之前对应于 A、B、C、D 的任何值(原始第二列)在相应列下排序 - A、B、C 或 D。必须保留原始顺序。

这是一个例子。

Input:

A|1
B|2
C|3
D|4
A|3
B|6
C|3
D|6

Desired output:
A|B|C|D
1|2|3|4
3|6|3|6

关于如何使用 Pandas/Python 完成此任务有什么想法吗？

非常感谢!

Answer 1

与两列旋转非常相似 ( Q/A 10 here ):

(df.assign(idx=df.groupby('col1').cumcount())
   .pivot(index='idx', columns='col1', values='col2')
)

输出:

col1  A  B  C  D
idx             
0     1  2  3  4
1     3  6  3  6