numpy.ma.argmax

ma.argmax(self, axis=None, fill_value=None, out=None)=<numpy.ma.core._frommethod object>

Возвращает массив индексов максимальных значений вдоль заданной оси. Маскированные значения обрабатываются так, как будто они имеют значение fill_value.

Параметры:

axis{None, целое число}

Если None, индекс в уплощенном массиве, в противном случае вдоль указанной оси

fill_valueскаляр или None, необязательно

Значение, используемое для заполнения маскированных значений. Если None, используется результат maximum_fill_value(self._data).

out{None, массив}, необязательно

Массив, в который может быть помещен результат. Его тип сохраняется, и он должен иметь правильную форму для хранения выходных данных.

Возвращает:

index_array{массив целых чисел}

Примеры

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