我想在一个numpy数组中找到所有最大值及其索引。
我正在寻找简单的东西。我开始像data [1:] 谢谢import numpy as np
data = np.array([0,1,20,1,0,1,2,30,2,1,0])
A = data[1:] < data[:-1]
您可以使用argsort
import numpy as np
data = np.array([0,1,20,1,0,1,2,30,2,1,0])
N = 2 # number of maximums
idx = np.argsort(data)[-N:]
idx
array([2, 7], dtype=int64)
data[idx]
array([20, 30])