a.argmax(axis=0)
array([1, 1, 0])
例子:
a = np.random.randint(50,size= (4,5))
print a
print a.argmax(axis=0)
output:
array:
[[34 7 11 35 47] [12 38 0 4 35] [32 11 46 21 25] [ 3 14 17 30 15]] inde: [0 1 2 0 0]
a.argmax(axis=0)
array([1, 1, 0])
例子:
a = np.random.randint(50,size= (4,5))
print a
print a.argmax(axis=0)
output:
array:
[[34 7 11 35 47] [12 38 0 4 35] [32 11 46 21 25] [ 3 14 17 30 15]] inde: [0 1 2 0 0]