检测数字列表中的峰值并记录它们的位置

我正在尝试创建一些代码来返回数值数组的“峰值”（或局部最大值）的位置和值。

例如，列表arr = [0, 1, 2, 5, 1, 0]在 位置有一个峰值3值为5 (since arr[3] equals 5).

数组的第一个和最后一个元素不会被视为峰值（在数学函数的上下文中，您不知道之后和之前的内容，因此，您不知道它是否是峰值）。

def pick_peaks(arr):
    print(arr)
    posPeaks = {
        "pos": [],
        "peaks": [],
    }
    startFound = False
    n = 0
    while startFound == False:
        if arr[n] == arr[n+1]:
            n += 1
        else:
            startFound = True

    endFound = False
    m = len(arr) - 1
    while endFound == False:
        if arr[m] == arr[m-1]:
            m -= 1
        else:
            endFound = True

    for i in range(n+1, m):
        if arr[i] == arr[i-1]:
            None
        elif arr[i] >= arr[i-1] and arr[i] >= arr[i+1]:
            posPeaks["pos"].append(i)
            posPeaks["peaks"].append(arr[i])

    return posPeaks

我的问题是平台期。[1, 2, 2, 2, 1]有一个峰值，同时[1, 2, 2, 2, 3]才不是。当平台是峰值时，记录平台的第一个位置。

任何帮助表示赞赏。

我建议你使用groupby https://docs.python.org/3/library/itertools.html#itertools.groupby将连续的相等值分组，然后为每个组存储第一个位置，例如[1, 2, 2, 2, 1]它在元组列表之后创建以下列表[(1, 0), (2, 1), (1, 4)]，将所有内容放在一起：

from itertools import groupby


def peaks(data):
    start = 0
    sequence = []
    for key, group in groupby(data):
        sequence.append((key, start))
        start += sum(1 for _ in group)

    for (b, bi), (m, mi), (a, ai) in zip(sequence, sequence[1:], sequence[2:]):
        if b < m and a < m:
            yield m, mi


print(list(peaks([0, 1, 2, 5, 1, 0])))
print(list(peaks([1, 2, 2, 2, 1])))
print(list(peaks([1, 2, 2, 2, 3])))

Output

[(5, 3)]
[(2, 1)]
[]