Разделить столбец на неизвестное количество столбцов с помощью разделителя панд

Question

Я пытаюсь разбить столбец на несколько столбцов на основе разделения запятой / пробела.

мой фрейм данных выглядит как

    Item                                          Colors
0   ID-1                                          Red, Blue, Green
1   ID-2                                          Red, Blue
2   ID-3                                          Blue, Green
3   ID-4                                          Blue
4   ID-5                                          Red

Я хотел бы преобразовать столбец «Цвета» в красный, синий и зеленый, например:

    Item                                           Red  Blue  Green
0   ID-1                                           1    1     1
1   ID-2                                           1    1     0
2   ID-3                                           0    1     1
3   ID-4                                           0    1     0
4   ID-5                                           1    0     1

Я действительно понятия не имею, как это сделать. Любая помощь будет принята с благодарностью.

Answer 1

Вы можете использовать get_dummies

pd.concat([df,df.Colors.str.get_dummies(sep=', ')],1)
Out[450]: 
   Item          Colors  Blue  Green  Red
0  ID-1  Red,Blue,Green     1      1    1
1  ID-2        Red,Blue     1      0    1
2  ID-3      Blue,Green     1      1    0
3  ID-4            Blue     1      0    0
4  ID-5             Red     0      0    1