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

Question

我正在将一个变量分类为一个新变量。但是，我无法在脚本中使用However函数：

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)

最后，我想将属性would = 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)