Python numpy MaskedArray.argmax用法及代碼示例

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

用法:

ma.MaskedArray.argmax(axis=None, fill_value=None, out=None, *, keepdims=<no value>)

返回沿給定軸的最大值的索引數組。屏蔽值被視為具有值 fill_value。

參數：

axis： {無，整數}

如果為 None，則索引在展平數組中，否則沿指定軸

fill_value： 標量或無，可選

用於填充掩碼值的值。如果沒有，則使用maximum_fill_value(self._data) 的輸出。

out： {無，數組}，可選

可以放置結果的數組。它的類型被保留，它必須具有正確的形狀來保存輸出。

返回：

index_array： {integer_array}

例子：

>>> a = np.arange(6).reshape(2,3)
>>> a.argmax()
5
>>> a.argmax(0)
array([1, 1, 1])
>>> a.argmax(1)
array([2, 2])