numpy.ma.choose

ma.choose(indices, choices, out=None, mode='raise')[源代码]

使用索引数组从一组选项中构造新数组。

给定一个整数数组和一组n个选择数组,此方法将创建一个合并每个选择数组的新数组。其中一个值 a 是i,新数组将具有选择的值 [i] 包含在同一位置。

参数
a铟锡合金

此数组中必须包含整数 [0, n-1] ,其中n是选项数。

choices阵列序列

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

out可选数组

如果提供,则结果将插入此数组。它应该是合适的形状和 dtype .

mode'raise'、'wrap'、'clip',可选

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

  • “raise”:引发错误

  • “环绕”:环绕

  • “剪辑”:剪辑到范围

返回
merged_array数组

参见

choose

等效函数

实例

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