Python numpy ma.choose用法及代碼示例

本文簡要介紹 python 語言中 numpy.ma.choose 的用法。

用法:

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

使用索引數組從選項列表中構造一個新數組。

給定一個整數數組和一個包含 n 個選擇數組的列表，此方法將創建一個新數組來合並每個選擇數組。當索引中的值為 i 時，新數組將具有choices[i] 在同一位置包含的值。

參數：

indices： 整數數組

此數組必須包含 [0, n-1] 中的整數，其中 n 是選擇的數量。

choices： 數組序列

選擇數組。索引數組和所有選擇應該可以廣播到相同的形狀。

out： 數組，可選

如果提供，結果將被插入到這個數組中。它應該具有適當的形狀和 dtype 。

mode： {‘raise’, ‘wrap’, ‘clip’}，可選

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

• ‘raise’：引發錯誤

• ‘wrap’：環繞

• ‘clip’：剪輯到範圍

返回：

merged_array： 數組

例子：

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