博客

np.select(condlist, choicelist, default=0)

condlist：条件列表，元素是bool数组

choicelist：列表，数组元素

default：当所有条件都不满足时，用default值填上。

condlist, choicelist的长度必须一致，每一个条件都要进行一次元素选择或者操作。

举例：

arr = np.arange(9).reshape(3,3)
conds = [arr < 3, 6 < arr]
np.select(condlist=conds, choicelist=[arr, arr*2])
array([[ 0,  1,  2],
       [ 0,  0,  0],
       [ 0, 14, 16]])

• 通俗讲，condlist是两个bool mask，choicelist是与bool mask形状相同的数组。正常情况下，condlist中所有的数组True与False应该是互补的，共同组成形状。但是，也可以重复，这时就以第一个True条件为准，如下：

conds = [arr < 8, 6 < arr]
np.select(condlist=conds, choicelist=[arr, arr*2])
array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7, 16]])

• condlist和choicelist中出现的数组可以不是同一个数组，但是。他们的形状必须形同。

arr = np.arange(9).reshape(3,3)
arr2 = np.arange(10,19).reshape(3,3)
conds = [arr < 3, 6 < arr]
np.select(condlist=conds, choicelist=[arr2, arr2*2])
array([[10, 11, 12],
       [ 0,  0,  0],
       [ 0, 34, 36]])