Добавление столбца в фрейм данных с логическим ответом - PullRequest
0 голосов
/ 06 октября 2018

Мой фрейм данных выглядит следующим образом:

In [16]: df
Out[16]:
   name  dep
0  jai  sales
1  bobby  design
2  ram  dev
3 shyam dev design

Мое намерение состоит в том, чтобы переделать фрейм данных следующим образом

In [16]: new_df
Out[16]:
   name  dev design sales 
0  jai 0 1 0
1  bobby  0 1 0
2  ram  1 1 0

Я попробовал следующее

new_df["dev"] = df[df["dep"].str.contains('dev')]
new_df["design"] = df[df["dep"].str.contains('design')]

Ноэто дает мне весь dataframe.Как получить логический результат?

Примечание: pd.get_dummies не дает мне такой результат

1 Ответ

0 голосов
/ 06 октября 2018

Используйте pd.get_dummies как:

df[['female','male']] = pd.get_dummies(df.gender)
del df['gender']

print(df)
    name  female  male
0    jai       0     1
1  bobby       1     0
2    ram       0     1

print(pd.get_dummies(df.gender))
   female  male
0       0     1
1       1     0
2       0     1

РЕДАКТИРОВАТЬ: Если в значении несколько строк, используйте:

df = df.join(df.dep.str.get_dummies(sep=' '))
del df['dep']

print(df)    
   design  dev  sales
0       0    0      1
1       1    0      0
2       0    1      0
3       1    1      0

print(df.dep.str.get_dummies(sep=' '))
   design  dev  sales
0       0    0      1
1       1    0      0
2       0    1      0
3       1    1      0
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...