当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python numpy select用法及代码示例


本文简要介绍 python 语言中 numpy.select 的用法。

用法:

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

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

参数

condlist bool ndarrays 列表

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

choicelist ndarrays 列表

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

default 标量,可选

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

返回

output ndarray

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

例子

>>> 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])

相关用法


注:本文由纯净天空筛选整理自numpy.org大神的英文原创作品 numpy.select。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。