numpy.ma.choose#

ma.choose(indices, choices, out=None, mode='raise')[source]#

使用索引数组从选择列表中构造新数组。

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

参数:

indicesndarray of ints（整数 ndarray）

此数组必须包含 [0, n-1] 范围内的整数，其中 n 是选择的数量。

choicessequence of arrays（数组序列）

选择数组。索引数组和所有选择应能广播到相同的形状。

outarray, optional（数组，可选）

如果提供，结果将插入到此数组中。它应具有适当的形状和 dtype。

mode{‘raise’, ‘wrap’, ‘clip’}, optional（{‘raise’, ‘wrap’, ‘clip’}，可选）

指定越界索引的行为方式。

‘raise’ : 抛出错误
‘wrap’ : 循环（wrap around）
‘clip’ : 裁剪到范围

返回:

merged_arrayarray（数组）

另请参阅

choose: 等效函数

示例

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