np.select(condlist, choicelist, default=0)
condlist是首选条件,choicelist是候选的条件.(如果condlist满足那么choicelist就没用),default是两者都满足后还剩余的默认值.
例1:
t = np.arange(6)
t
Out[12]: array([0, 1, 2, 3, 4, 5])np.select([t<6,t>2,True],[t+10,t-10,66]) #因为t<6都满足,索引执行t+10,t>2和默认值都不执行
Out[11]: array([10, 11, 12, 13, 14, 15])
例2:
np.select([t>3,t<1,True],[t+10,t+20,66]) #t>3时执行t+10,t<1时执行t+20,剩余的执行66
Out[14]: array([20, 66, 66, 66, 14, 15])