我可以比较列表的一部分吗？

Question

此代码将选择列表“a”的 6 个或多个正值

a=np.array([1.01, -1.58, 0.64, 1.38, 0.69, 0.91, 1.34, 1.03, 1.39, 0.94, -1.01,0.16])
b= np.zeros(13)
for i in range(6):
    if (a[i] > 0 and a[i+1] > 0 and a[i+2] > 0 and a[i+3] > 0\
        and a[i+4] > 0 and a[i+5] > 0 and a[i+6] > 0) :
            
            b[i] = a[i]
            b[i+1] = a[i+1]
            b[i+2] = a[i+2]
            b[i+3] = a[i+3]
            b[i+4] = a[i+4]
            b[i+5] = a[i+5]
            b[i+6] = a[i+6]

我想让它更短，我试过了：

for i in range (6):
    if (a[i:i+6]) > 0:
        b[i:i+6] = a[i:i+6]

Answer 1

重新审视这个 - 我想添加这种替代方法，其中包括我在 previous answer 中提到的优化：

这个版本要长得多，并且需要手动编码跟踪每个正数系列的索引。但它具有主要的性能优势，如下所述。

a = np.array([1.01, -1.58, 0.64, 1.38, 0.69, 0.91, 1.34, 1.03, 1.39, 0.94, -1.01, 0.16,])
b = np.zeros(len(a))

MIN_CONSECUTIVE = 6  # number of consecutive numbers wanted
first, last, c = -1, -1, 0  # index of first & last positive number, count of +ve nums
nums = []  # hold each series of consecutive positive numbers

for i, n in enumerate(a):
    if n > 0:
        nums.append(n)
        first = i if first == -1 else first
        last = i
        c += 1
    else:
        if c >= MIN_CONSECUTIVE:
            b[first:last+1] = nums
        first, last, c = -1, -1, 0  # reset
        nums = []  # reset
        if i > len(a) - MIN_CONSECUTIVE - 1:  # shortcut exit if there aren't
            break                             # enough elems left for consecutive

else:  # if the loop completes without `break`
    if c >= MIN_CONSECUTIVE:  # also check after the loop completes
        b[first:last+1] = nums

b  # -> array([0.  , 0.  , 0.64, 1.38, 0.69, 0.91, 1.34, 1.03, 1.39, 0.94, 0.  , 0.  ])

为什么这样更好：

1. 这个访问和访问 numpy 数组（或 python 列表）中的每个项目恰好一次。
   • 您尝试并在 cmets 和我之前的回答中提供的方法多次查找 重叠 数组中的一系列元素
   • 前一个也多次重写到b，并用相同的值覆盖
2. 只有在找到并完成每个系列后才会写入 b。
3. 我正在使用 enumerate() 跟踪索引，但我不使用索引来访问元素。 python 列表上的索引访问比仅遍历列表的元素要慢。 The same is true for numpy arrays.
4. 对于您给定的列表，此列表的运行时间不到上一个答案的一半：
   • 这：14.6 µs ± 584 ns
   • 上一页：31.9 µs ± 1.13 µs

Answer 2

在缩短的版本中，您需要更改的主要内容是if (a[i:i+6]) > 0:。此外，a 的长度为 12，b 的长度为 13。编辑 1：（根据@Steve's comment below）此外，索引range 循环也可以从for i in range(6) 推广到for i in range(len(a)-5)So：

b = np.zeros(len(a))
# b is 
# array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.])
for i in range(len(a)-5):
    if all(a[i:i+6] > 0):
        b[i:i+6] = a[i:i+6]

b
# output:
# array([0.  , 0.  , 0.64, 1.38, 0.69, 0.91, 1.34, 1.03, 1.39, 0.94, 0.  , 0.  ])

编辑2，按照@hpaulj's comment，因为a是一个numpy数组，我们不需要遍历切片，因为numpy会将>运算符应用于数组的每个元素（切片）并返回[True, False, ...] 的数组。第一个循环看起来像这样：

例如a[0:0+6] > 0 返回array([ True, False, True, True, True, True])。

---

一次准确检查 6 个元素的一个显着缺点是，如果 是一组较长的连续正数，它将在下一个循环中填充，并覆盖 b 中已经存在的相同值。所以不是很优化。对索引进行显式检查可能会更好，直到您获得非正数（或结尾）并确保您的计数当前 > 6。