So I know that the numpy argmax retrieves the maximum value along an axis. Thus,
x = np.array([[12,11,10,9],[16,15,14,13],[20,19,18,17]])
print(x)
print(x.sum(axis=1))
print(x.sum(axis=0))
would output,
[[12 11 10 9]
[16 15 14 13]
[20 19 18 17]]
[42 58 74]
[48 45 42 39]
This makes sense as the sum along axis 1 (row) is [42 58 74]
and axis 0 (column) is [48 45 42 39]
.
However, i am confused of how argmax work. From my understanding, argmax is supposed to return the max number along the axis. Below is my code and output.
Code: print(np.argmax(x,axis=1))
. Output: [0 0 0]
Code: print(np.argmax(x,axis=0))
. Output: [2 2 2 2]
Where does 0
and 2
come from? I've deliberately used a set of more complex integer values (9..20) so as to distinguish between the 0
and 2
and the integer values inside the array.
See Question&Answers more detail:
os 与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…