Транспонировать значения столбца dataframe как заголовки столбцов в python - PullRequest
0 голосов
/ 23 мая 2019

Кадр данных pandas имеет три столбца (HHID, Имя , модель). Я хочу вывести значения столбцов модели в качестве заголовков столбцов. Значения в «модели» должны быть «Да», если естьзначение модели, соответствующее столбцу HHID.

Может кто-нибудь помочь мне с кодом Python для этого.

Прикрепление снимка экрана ввода.

Input screenshot

Выходные данные должны быть:

Output Screenshot

1 Ответ

0 голосов
/ 23 мая 2019

Вы можете попробовать pd.get_dummies

     hhid _NAME_    model
0   10006  flag4      NaN
1   10014  flag4      NaN
2   10047  flag4  make323
3   10055  flag4      NaN
4   10066  flag4  make268
5   10074  flag4      NaN
6   10085  flag4      NaN
7   10117  flag4  make146
8   10126  flag4      NaN
9   10136  flag4      NaN
10  10147  flag4  make148
>>> pd.get_dummies(df, columns=['model'])

     hhid _NAME_  model_make146  model_make148  model_make268  model_make323
0   10006  flag4              0              0              0              0
1   10014  flag4              0              0              0              0
2   10047  flag4              0              0              0              1
3   10055  flag4              0              0              0              0
4   10066  flag4              0              0              1              0
5   10074  flag4              0              0              0              0
6   10085  flag4              0              0              0              0
7   10117  flag4              1              0              0              0
8   10126  flag4              0              0              0              0
9   10136  flag4              0              0              0              0
10  10147  flag4              0              1              0              0
>>> # by setting prefix and prefix_sep to empty string
>>> # you could make your values as column names
>>> pd.get_dummies(df, columns=['model'], prefix='', prefix_sep='')

     hhid _NAME_  make146  make148  make268  make323
0   10006  flag4        0        0        0        0
1   10014  flag4        0        0        0        0
2   10047  flag4        0        0        0        1
3   10055  flag4        0        0        0        0
4   10066  flag4        0        0        1        0
5   10074  flag4        0        0        0        0
6   10085  flag4        0        0        0        0
7   10117  flag4        1        0        0        0
8   10126  flag4        0        0        0        0
9   10136  flag4        0        0        0        0
10  10147  flag4        0        1        0        0
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...