NumPy参考 >例行程序 >Indexing routines > numpy.select
numpy.
select
(condlist,选择列表,默认= 0 )[源] ¶根据条件返回从选择列表中的元素绘制的数组。
的其中确定从哪个数组中的条件列表选择列表 中所采取的输出元件。当满足多个条件时,将使用condlist中遇到的第一个条件。
从中获取输出元素的数组列表。它的长度必须与condlist相同。
当所有条件评估为False时,元素将插入输出中。
在位置m的输出是第m个所述阵列的在元件 选择列表,其中在相应的阵列的第m个元素 condlist为True。
例子
>>> x = np.arange(10)
>>> condlist = [x<3, x>5]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist)
array([ 0, 1, 2, ..., 49, 64, 81])