Извлечение символа из str и переформатирование фрейма данных с помощью Pandas - PullRequest
0 голосов
/ 26 мая 2018

У меня есть Pandas DataFrame, подобный этому:

df = pd.DataFrame({
    'Date': ['2017-1-1', '2017-1-2', '2017-1-3'],
    'Groups': [['one','two'], ['one','three'], 'four']})

    Date      Groups          
0  2017-1-1    one, two       
1  2017-1-2    one, three          
2  2017-1-3    four                

Как я могу сгенерировать новый DataFrame, подобный этому?

    Date      Groups          one     two     three   four
0  2017-1-1    one, two       1        1        0       0
1  2017-1-2    one, three     1        0        1       0      
2  2017-1-3    four           0        0        0       1    

1 Ответ

0 голосов
/ 26 мая 2018

join с get_dummies

df.join(pd.get_dummies(df.Groups.apply(pd.Series), prefix='', prefix_sep=''))

       Date        Groups  four  one  three  two
0  2017-1-1    [one, two]     0    1      0    1
1  2017-1-2  [one, three]     0    1      1    0
2  2017-1-3          four     1    0      0    0
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...