如何在np.select函数中使用else

Question

我将一个变量分类为一个新变量。但是，我无法在脚本中使用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

使用我使用的格式有什么方法可以做到这一点吗

Answer 1

为默认值添加第三个参数：

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)