Как использовать else в функции np.select - PullRequest
2 голосов
/ 04 августа 2020

Я помещаю переменную в новую переменную. Однако я не могу использовать функцию else в моем скрипте:

df['cod_cat'] = np.select([
(df['cod'] >= 223101) & (df['cod'] <= 223199),
(df['cod'] >= 223201) & (df['cod'] <= 223299),
(df['cod'] >= 223301) & (df['cod'] <= 223399),
(df['cod'] >= 223401) & (df['cod'] <= 223499),
(df['cod'] >= 223501) & (df['cod'] <= 223599),
(df['cod'] == 223605)],[1,2,3,4,5,6)

В конце я хотел бы поставить атрибут else = 99.

Есть ли способ сделать это с помощью формат, который я использовал?

1 Ответ

3 голосов
/ 04 августа 2020

Добавьте третий параметр для значения по умолчанию:

df['cod_cat'] = np.select([
(df['cod'] >= 223101) & (df['cod'] <= 223199),
(df['cod'] >= 223201) & (df['cod'] <= 223299),
(df['cod'] >= 223301) & (df['cod'] <= 223399),
(df['cod'] >= 223401) & (df['cod'] <= 223499),
(df['cod'] >= 223501) & (df['cod'] <= 223599),
(df['cod'] == 223605)], [1,2,3,4,5,6], default=99)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...