numpy.select #

麻木的。select ( condlist , choicelist ,默认= 0 ) [来源] #

根据条件返回从选择列表中的元素提取的数组。

参数
condlist bool ndarray 列表

条件列表,用于确定从choicelist中的哪个数组中 获取输出元素。当满足多个条件时,使用condlist中遇到的第一个条件。

ndarrays 的choicelist列表

从中获取输出元素的数组列表。它的长度必须与condlist相同。

默认标量,可选

当所有条件评估为 False 时插入到输出中的元素。

返回
输出数组

位置 m 处的输出是 choicelist中数组的第 m 个元素,其中condlist中相应数组的第 m 个元素 为 True。

也可以看看

where

根据条件从两个数组之一返回元素。

take, choose, compress, diag,diagonal

例子

>>> x = np.arange(6)
>>> condlist = [x<3, x>3]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist, 42)
array([ 0,  1,  2, 42, 16, 25])
>>> condlist = [x<=4, x>3]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist, 55)
array([ 0,  1,  2,  3,  4, 25])