查找值高于两侧的 w 元素的列表元素

问题描述

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

    A = [1,2,3,1,4,5,6,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,3,1)，它是左边 'w' 元素 1,2 和右边 'w' 元素 2,1 中的最大值。同样，第二个输出 4 来自 (2,4,2) 等等。

如何扩展 w = n，其中 n = 3,5 ？

解决方法

import numpy as np

---

如果你有：

A = np.array([1,2,3,1,4,5,6,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,6]

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

[9,11]

正如预期的那样，我希望。

,

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

A = [1,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])

输出是

3
4
5
6

,

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

输出（两者）：

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]

输出：

[3,6]