python - Pandas:转换数据框以显示原始数据框中是否存在值的组合

Question

我有一个如下所示的数据框:

 | Col 1 | Col 2 | 
0|   A   |   2   |
1|   A   |   3   |
2|   B   |   1   |
3|   B   |   2   |

我需要将其转换为一个数据帧，该数据帧显示第 1 列和第 2 列中的值的每个组合(如果该组合包含在原始数据帧中):

  |  1  |  2  |  3  |
A |False|True |True |
B |True |True |False|

pandas 是否有本地方法来实现这种转换？我手动创建转换后的数据框，但这很慢。

提前谢谢您!

Answer 1

您可以使用:

df.groupby(['Col 1','Col 2']).size().unstack(fill_value=0).astype(bool)

Col2      1     2      3
Col1                    
A     False  True   True
B      True  True  False