np.selcect()

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])