首页 > 解决方案 > Pandas:转换数据框以显示原始数据框中是否存在值组合

问题描述

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

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

如果该组合包含在原始DataFrame中,我需要将其转换为显示Col 1和Col 2中的值的每个组合的Dataframe:

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

pandas 中是否有一种本地方式来实现这种转换?我正在手动创建转换后的 Dataframe,但这会很慢。

先感谢您!

标签: pythonpandas

解决方案


你可以使用:

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

推荐阅读