在numpy数组上查找所有最大值

Posted 2023-01-26

我想在一个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])