argmax gibt die Elementnummer des Elements mit der größten durch die Achse angegebenen Abmessung aus.
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]
In print (np.argmax (Array, Achse = 0)),
[2,5,3,2] [3,6,2,1] [4,4,1,3] werden ins Visier genommen.
Daher werden "5", "6", "4" aufgenommen.
Jede Elementnummer ist [1 1 0].
In print (np.argmax (Array, Achse = 1)),
[2,3,4], [5,6,4], [3,2,1], [2,1,3] werden ins Visier genommen.
Daher werden "4", "6", "3", "3" aufgenommen.
Jede Elementnummer ist [2 1 0 2].
Wenn keine Achse angegeben ist, ist dies 4. Ich habe das aus irgendeinem Grund nicht untersucht.
tmp.py
print(np.argmax(array) #=> 4
Recommended Posts