查找比两边的 w 元素都高的列表元素

Question

这就是我能够为 w = 2 写的内容。

    A = [1,2,3,2,1,4,3,2,5,1,2,6,1,2,7]
    w = 2
    
    for i in range(w, len(A)- w):
        for j in range(w):
            if A[i] > A[j] and A[i] > A[i-1]:
                if A[i] > A[j+4] and A[i] > A[i+1]:
                    print(A[i])

Output: 3
        4
        5
        5
        6
        6

第一个输出是 3 (1,2,3,2,1) 这是它左边 'w' 元素 1,2 和右边 'w'元素2,1。同样，第二个输出 4 来自 (2,1,4,3,2) 等等。

如何将其扩展为 w = n，其中 n = 3、4、5？

Answer 1

import numpy as np

如果你有：

A = np.array([1,2,3,2,1,4,3,2,5,1,2,6,1,2,7])

你可以这样做：

result=[]
w=2
for i in range(w, len(A)-w):
    if all(A[i] > A[i-w:i]) and all(A[i] > A[i+1:i+w+1]):
        result.append(A[i])

result 将是：

[3, 4, 5, 6]

如果将 A 的定义更改为 A = np.array([7,2,-3,9,1,4,3,2,10,11,2,6,1,2,7])，将 w 更改为 3，则 result 将是：

[9, 11]

如期而至，希望。

Answer 2

虽然我很喜欢 zabop 的回答，但有一种更简单的方法可以达到相同的结果，而且无需使用 numpy:

A = [1,2,3,2,1,4,3,2,5,1,2,6,1,2,7]
w = 2

for i in range(w, len(A)- w):
    greater = True
    for j in range(1,w+1):
        if A[i-j] >= A[i] or A[i+j] >= A[i]:
            greater = False
            break
    if greater:
        print(A[i])

输出为

Answer 3

仅使用 Python 中的 for loop:

for i in range(w, len(A)-w):
    s = []
    for j in range(1, w+1):
        s.append(A[i] >= A[i-j] and A[i] >= A[i+j])
    if sum(s)==w:
        print(A[i])

混合 for loop 与 list comprehension:

for i in range(w, len(A)-w):
    if sum([A[i] >= A[i-j] and A[i] >= A[i+j] for j in range(1,w+1)]) == w:
        print(A[i])

输出（两者）：

疯狂单线list comprehension:

[A[i] for i in range(w, len(A)-w) if sum([A[i] >= A[i-j] and A[i] >= A[i+j] for j in range(1,w+1)]) == w]

输出：

[3, 4, 5, 6]

查找比两边的 w 元素都高的列表元素

Find list elements with higher values than the w elements on both sides

python

max

numerical-methods