Я не могу понять вывод argmax
и argmin
при использовании с параметром оси. Например:
>>> a = np.array([[1,2,4,7], [9,88,6,45], [9,76,3,4]])
>>> a
array([[ 1, 2, 4, 7],
[ 9, 88, 6, 45],
[ 9, 76, 3, 4]])
>>> a.shape
(3, 4)
>>> a.size
12
>>> np.argmax(a)
5
>>> np.argmax(a,axis=0)
array([1, 1, 1, 1])
>>> np.argmax(a,axis=1)
array([3, 1, 1])
>>> np.argmin(a)
0
>>> np.argmin(a,axis=0)
array([0, 0, 2, 2])
>>> np.argmin(a,axis=1)
array([0, 2, 2])
Как вы можете видеть, максимальное значение - это точка (1,1), а минимальная - точка (0,0). Поэтому в моей логике, когда я запускаю:
-
np.argmin(a,axis=0)
Я ожидалarray([0,0,0,0])
-
np.argmin(a,axis=1)
Я ожидалarray([0,0,0])
-
np.argmax(a,axis=0)
Я ожидалarray([1,1,1,1])
-
np.argmax(a,axis=1)
Я ожидалarray([1,1,1])
Что не так с моим пониманием вещей?