python - 如何在 Pandas 中枚举和拆分 csv 列？

Question

我们有一个 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 列？

标签： pythonpandascsv

Answer 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())