numpy.ma.choose

ma.choose(indices, choices, out=None, mode='raise')

使用索引数组从一系列选择中构造一个新的数组.

给定一个整数数组和 n 个选择数组的列表,此方法将创建一个合并每个选择数组的新数组. 当 index 中的值为 i 时,新数组将具有 choices[i] 在相同位置包含的值.

参数:

indices整数类型的ndarray

此数组必须包含 [0, n-1] 中的整数,其中 n 是选择的数量.

choices数组序列

选择数组. 索引数组和所有选择都应可广播到相同的形状.

outarray,可选

如果提供,结果将被插入到此数组中. 它应该是适当的形状和 dtype .

模式{‘raise’, ‘wrap’, ‘clip’},可选

指定越界索引的行为.

• ‘raise’ : 引发错误

• ‘wrap’ : 环绕

• ‘clip’ : 裁剪到范围

返回:

merged_arrayarray

参见

choose

等效函数

示例

Try it in your browser!

>>> import numpy as np
>>> choice = np.array([[1,1,1], [2,2,2], [3,3,3]])
>>> a = np.array([2, 1, 0])
>>> np.ma.choose(a, choice)
masked_array(data=[3, 2, 1],
             mask=False,
       fill_value=999999)

Go BackOpen In Tab