py: np.argmax()

https://blog.csdn.net/weixin_38145317/article/details/79650188

1.對一個一維向量

import numpy as np
a = np.array([3, 1, 2, 4, 6, 1])
b=np.argmax(a)#取出a中元素最大值所對應的索引，此時最大值位6，其對應的位置索引值爲4，（索引值默認從0開始）
print(b)#4

2.對2維向量（通常意義下的矩陣）a[][]

import numpy as np
a = np.array([[1, 5, 5, 2],
              [9, 6, 2, 8],
              [3, 7, 9, 1]])
b=np.argmax(a, axis=0)#對二維矩陣來講a[0][1]會有兩個索引方向，第一個方向爲a[0]，默認按列方向搜索最大值
#a的第一列爲1，9，3,最大值爲9，所在位置爲1，
#a的第一列爲5，6，7,最大值爲7，所在位置爲2，
#此此類推，因爲a有4列，所以得到的b爲1行4列，
print(b)#[1 2 2 1]
 
c=np.argmax(a, axis=1)#現在按照a[0][1]中的a[1]方向，即行方向搜索最大值，
#a的第一行爲1，5，5，2,最大值爲5（雖然有2個5，但取第一個5所在的位置），索引值爲1，
#a的第2行爲9，6，2，8,最大值爲9，索引值爲0，
#因爲a有3行，所以得到的c有3個值，即爲1行3列
print(c)#[1 0 2]