numpy.ma.choose

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

使用索引陣列從選項列表中建構新陣列。

給定一個整數陣列和 n 個選項陣列的列表，此方法將建立一個新陣列，合併每個選項陣列。當 index 中的值為 i 時，新陣列在相同位置將具有 choices[i] 中包含的值。

參數:

indices整數的 ndarray

此陣列必須包含 [0, n-1] 範圍內的整數，其中 n 是選項的數量。

choices陣列序列

選項陣列。索引陣列和所有選項應可廣播為相同形狀。

out陣列，選用

如果提供，結果將插入到此陣列中。它應具有適當的形狀和 dtype。

mode{‘raise’, ‘wrap’, ‘clip’}，選用

指定超出邊界索引的行為方式。

• ‘raise’：引發錯誤

• ‘wrap’：環繞

• ‘clip’：裁剪到範圍

回傳值:

merged_array陣列

另請參閱

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)