Использование numpy.argmax

Question

Я здесь, чтобы узнать об использовании numpy.argmax

Например, рассмотрим этот массив:

import numpy as np

a = np.arange(6).reshape(2,3)
b = np.argmax(a, axis = 0)
c = np.argmax(a, axis = 1)

print(a)
print(b)
print(c)

Вот вывод:

[[0 1 2]
 [3 4 5]]
5
[1 1 1]
[2 2]

Я запутался в использовании оси параметров для numpy.argmax.Что оно делает?Почему он возвращает [1 1 1], если ось = 0, и [2 2], если значение оси = 1?

Answer 1

argmax возвращает вам индекс максимального значения вдоль указанной вами оси.

Точные сравнения, которые он сделал для этого:

3> 0, 4> 1, 5>2: [1 1 1]

2 является самым большим из множества [0 1 2] 5 является самым большим из множества [3 4 5]:

[2 2]

Answer 2

numpy.argmax() возвращает позицию самого большого элемента в массиве, необязательно по строке или столбцу (аргумент axis).Таким образом, в первом случае, [1 1 1], вы получаете позицию наибольшего элемента по столбцам.Поскольку все элементы в строке 1 больше, чем элементы в строке 0, вы получите массив из трех элементов.Аналогично для axis=1, где вы получаете столбец самого большого элемента в каждой строке.