如何通过Python转换.CSV标签？

Question

我有一个 .csv 标签并有四个不同的类别。

现在我的 .csv 文件如下所示：

id type

1 1

2 2

3 3

4 4

5 2

...

我想将其转换为：

id type1 type2 type3 type4

1 1 0 0 0

2 0 1 0 0

3 0 0 1 0

4 0 0 0 1

5 0 1 0 0

我怎样才能通过python完成这些？我使用 pd.read_csv()

Answer 1

如果您的 csv 文件（此处命名为 file.csv）看起来像这样

id,type
1,1
2,2
3,3
4,4
5,2

，那么您可以使用 .str.get_dummies() 获取

df = (
    pd.read_csv("file.csv", index_col=0)
    .type.astype(str).str.get_dummies().rename(lambda c: f"type{c}", axis=1)
)

以下数据帧

    type1  type2  type3  type4
id                            
1       1      0      0      0
2       0      1      0      0
3       0      0      1      0
4       0      0      0      1
5       0      1      0      0

如果您想将其写回一个新的 csv 文件，然后

df.to_csv("file_new.csv", index=True)

生成以下文件 file_new.csv：

id,type1,type2,type3,type4
1,1,0,0,0
2,0,1,0,0
3,0,0,1,0
4,0,0,0,1
5,0,1,0,0