argmax renvoie le numéro d'élément de celui dont la dimension est la plus grande spécifiée par axis.
tmp.py
import numpy as np
array = [[2,3,4],
[5,6,4],
[3,2,1],
[2,1,3]]
array = np.array(array)
print(np.argmax(array, axis=0)) # [1 1 0]
print(np.argmax(array, axis=1)) # [2 1 0 2]
Dans print (np.argmax (array, axis = 0)),
[2,5,3,2] [3,6,2,1] [4,4,1,3] sont ciblés.
Par conséquent, "5", "6", "4" sont repris.
Chaque numéro d'élément est [1 1 0].
Dans print (np.argmax (array, axis = 1)),
[2,3,4], [5,6,4], [3,2,1], [2,1,3] sont visés.
Par conséquent, "4", "6", "3", "3" sont captés.
Chaque numéro d'élément est [2 1 0 2].
Si l'axe n'est pas spécifié, ce sera 4. Je n'ai pas enquêté sur cela pour une raison quelconque.
tmp.py
print(np.argmax(array) #=> 4
Recommended Posts