python---之求每一行或每一列的最大值的索引（轴）

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]