python - Pandas 中的可旋转

Question

我有一个像这样的数据框:

Col1 Col2
A    Foo1
A    Foo3
B    Foo1
B    Foo3
B    Foo4
C    Foo1
D    Foo3
D    Foo4

我需要这样的输出:

       Foo1     Foo3      Foo4
A    1             1            0
B    1             1            1
C    1             0            0
D    0             1            1

在 pandas 中实现这一点的最简单方法是什么，而不使用 python 中的 foreach 循环。

Answer 1

最简单的是crosstab :

df1 = pd.crosstab(df['Col1'], df['Col2'])

另一个解决方案是 GroupBy.size与 unstack :

df1 = df.groupby(['Col1','Col2']).size().unstack(fill_value=0)

print (df1)
Col2  Foo1  Foo3  Foo4
Col1                  
A        1     1     0
B        1     1     1
C        1     0     0
D        0     1     1