numpy.select

原文：https://docs.scipy.org/doc/numpy/reference/generated/numpy.select.html

译者：飞龙 UsyiyiCN

校对：（虚位以待）

numpy.select(condlist, choicelist, default=0)[source]

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

参数：	condlist：bool ndarrays列表 确定从选择列表中的哪个数组取得输出元素的条件列表。当满足多个条件时，使用condlist中遇到的第一个条件。 choicelist：ndarrays列表 从中获取输出元素的数组的列表。它的长度必须与condlist相同。 默认：标量，可选 当所有条件求值为False时，在输出中插入的元素。
返回：	输出：ndarray 位置m处的输出是选择列表中数组的第m个元素，其中condlist中对应数组的第m个元素为True。

也可以看看

where

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

take，choose，compress，diag，diagonal

例子

>>> x = np.arange(10)
>>> condlist = [x<3, x>5]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist)
array([ 0,  1,  2,  0,  0,  0, 36, 49, 64, 81])