查找比两边的 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

第一个 output 是 3 (1,2, 3 ,2,1)，它是左“w”元素 1,2 和右“w”元素 2,1 中的最大值。 同样，第二个 output 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 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])

output 是

3
4
5
6

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])

Output（两者都适用）：

3
4
5
6

---

一个疯狂的单行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]

Output：

[3, 4, 5, 6]