如何将分类列值转换为不同的列名？

Question

我不知道如何解决这个问题，因为我是 pandas 的初学者。

我有这个数据框：

  col1 col2
0    a    1 
1    a    2 
2    a    3 
3    b    4
4    b    5
5    b    6
6    c    7
7    c    8
8    c    9

我想把它变成这样的数据框或矩阵：

   cola colb  colc
0    1    4    7
1    2    5    8
2    3    6    9

我应该如何在 Python 中解决这个问题？

Answer 1

让我们 groupby col1 上的数据框并在 dict 理解中创建键值对：

pd.DataFrame({k: [*g['col2']] for k, g in df.groupby('col1')})

或者，您可以使用groupby + cumcount 创建一个顺序计数器来区分col1 中每个组的不同行，然后使用set_index + unstack 来重塑：

df.set_index([df.groupby('col1').cumcount(), 'col1'])['col2'].unstack()

pivot_table 和 groupby + cumcount 的另一种方法：

df.pivot_table(index=df.groupby('col1').cumcount(), columns='col1', values='col2')

结果：

   a  b  c
0  1  4  7
1  2  5  8
2  3  6  9