numpy.select¶
-
numpy.
select
(condlist, choicelist, default=0)[source]¶ 返回根据条件从选择列表中的元素绘制的数组。
参数: condlist:bool ndarrays列表
确定从选择列表中的哪个数组取得输出元素的条件列表。当满足多个条件时,使用condlist中遇到的第一个条件。
choicelist:ndarrays列表
从中获取输出元素的数组的列表。它的长度必须与condlist相同。
默认:标量,可选
当所有条件求值为False时,在输出中插入的元素。
返回: 输出:ndarray
位置m处的输出是选择列表中数组的第m个元素,其中condlist中对应数组的第m个元素为True。
例子
>>> 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])