如何枚举和拆分熊猫中的csv列?

时间:2020-06-29 00:16:20

标签: python pandas csv

我们有一个CSV。它在列中具有一组固定的(重复的)字符串。说列称为“类型”,值是[A, B, B, C]。我想获得几个新列,它们等于唯一列值的数量,其中包含0或1。像这样:

Type Type_1 Type_2 Type_3
A      1      0      0
B      0      1      0
B      0      1      0
C      0      0      1

如何将一列变成一组0 \ 1列?

1 个答案:

答案 0 :(得分:1)

尝试使用str.get_dummies

df.Type.str.get_dummies()
   A  B  C
0  1  0  0
1  0  1  0
2  0  1  0
3  0  0  1

更新

df=df.join(df.pop('Type').str.get_dummies())