根据条件添加几个标志列

时间:2019-10-24 07:53:47

标签: python pandas dataframe

我有一个这样的数据框-

id  colors
1   red,blue,yellow,black
2   blue, yellow
3   black, orange
4   red, black, orange
5   blue, yellow, orange

我想添加三个标志columnms来指示是否存在该颜色。

我得到的数据框应如下所示-

id  colors                      blue    black   orange
1   red,blue,yellow,black        1        1       0
2   blue, yellow                 1        0       0
3   black, orange                0        1       1
4   red, black, orange           0        1       1
5   blue, yellow, orange         1        0       1

0 个答案:

没有答案