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

Question

У меня есть столбец в кадре данных pandas (с использованием PyCharm) со значениями:

Colour
-----
Red
Green
Green
Blue
Red

Но я бы хотел разделить его на несколько столбцов с заголовками «Красный», «Зеленый», «Синий»"и затем установите значения" да "или" нет ".

Red  Blue Green
--------------
Yes  No  No
No   No  Yes
No   No  Yes
No   Yes No
Yes  No  No

Answer 1

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

df = pd.DataFrame({'Colour': ['Red', 'Green', 'Green', 'Blue', 'Red']})
df
    Colour
0   Red
1   Green
2   Green
3   Blue
4   Red

pd.get_dummies(df['Colour']).replace({0:'No', 1:'Yes'})
     Blue            Green  Red
0     No              No    Yes
1     No              Yes   No
2     No              Yes   No
3     Yes             No    No
4     No              No    Yes

Answer 2

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

df.Colour.str.get_dummies().replace({1:'Yes', 0:'No'})

   Blue Green Red
0   No    No  Yes
1   No   Yes   No
2   No   Yes   No
3  Yes    No   No
4   No    No  Yes