如何在列表中查找和替换间隔的值？

Question

我有一个这样的列表：

list_1 = [1, 2, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 1, 2, 1, 2, 3, 1, 1, 2, 3, 4, 1]

在此列表中，紧随4 之后的值大于或小于4 本身。更具体地说，较小的值始终是1。从那个1 到下一个1 的间隔值总是小于4。

如何查找和替换这样一个区间的值，例如如下：

list_2 = [nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, 0, 0, 0, nan, nan, nan, nan, nan, nan, nan, 0]

1, 2, 1 被替换，因为第一个 1 就在 4 之后并且小于 4。因此，从第一个 1 到第二个 1 的间隔被替换。

其他值不必是nan，我只是突出显示替换。需要明确的是，如果后面的值大于4，我们将跳过它。

Answer 1

作为一个选项：

for i in range(len(list_1)): # looking at all the elements in the list in order.
    if a[i]==4: # if the number with "i" index equals to "4"
        if a[i+1] > 4: # if the number after 4 is greater than 4
            a[i+1] == nan # the number after 4 equals "nan"

Answer 2

list1 = [1, 2, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 1, 2, 1, 2, 3, 1, 1, 2, 3, 4, 1]
list2 = []
i = 0
while i != len(list1):
    if list1[i] == 4:
        list2.append("nan")
        if list1[i+1] > 4:
            list2.append("Superior to 4 after a 4")
            i+=1
        else:
            list2.append("nan")
            i+=1
    else:
        list2.append("nan")
    i += 1
print(list2)

代码没有优化，但是如果对应的值不优于4，则在4后面加上“nan”，如果是，则在4后面加上“superior to 4”。

Answer 3

这是我让 numpy 完成大部分繁重工作的尝试，
不幸的是，我最后确实使用了 for 循环，希望有人可以提出建议并进行编辑以获得更好的解决方案。

这是代码：

import numpy as np
list_1 = np.array([1, 2, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 1, 2, 1, 2, 3, 1, 1, 2, 3, 4, 1])

modified_list_1 = np.append(list_1,1) # to always have a "next 1"

idx_of_4 = np.where(modified_list_1==4)[0]
idx_of_1 = np.where(modified_list_1==1)[0]

idx_of_4_followed_by_1 = np.intersect1d(idx_of_4, idx_of_1-1)

arr_slice_idx = [(start, np.min(idx_of_1[idx_of_1>(start+1)])) for start in idx_of_4_followed_by_1]

for start,end in arr_slice_idx:
  list_1[start+1:end+1] = 0

print(list_1)

我首先使用np.where 和np.intersect1d 查找4s 的索引，然后是1s，这是矢量化的，应该可以非常快速地工作
不幸的是，我在这里没有灵感了，为了找到关闭每个范围的“下一个1”，我使用了常规（相当丑陋）的理解。

然后当我有了开始和结束时，我使用它们对原始数组进行切片并将值设置为0。