给定数组
import numpy as np
A = np.array([[1,2],[3,4]])
如果想要找到最大值可以使用
max_A = np.max(A)
进一步,如果想要得到最大值所在的位置,根据英语,感觉argmax就可以做到
A.argmax(max_A)
#>>> 3
返回值是3,这是由于在python存储数组时可以理解为存成一个列向量加上一个形状,所以直接使用index返回的索引就是展开成列向量数的位置。
我们希望它返回的值是[1,1]
这个时候就需要用到where了
np.where(A == np.max(A))
其中A == np.max(A)返回的是一个mask,结合where就可以解决问题了。
网友评论