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

问题描述 投票:0回答:1

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

谢谢

python-3.x numpy
1个回答
0
投票

您可以使用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])
© www.soinside.com 2019 - 2024. All rights reserved.