检测和打印条纹

Question

我写了一些代码来检测数据中的条纹，我只是想知道为什么它不工作。

我得到了-1，0和1的序列。我想报告一下非零值的位置。但它们往往是以条纹的形式出现的；因此，我想要报告的不是单个位置，而是条纹。下面是一个简单的例子：

sequence=[-1,-1,0,0,1,1]

print('Negative Influences:')
streaking=False
streak=[]
for j in range(len(sequence)):
    if streaking==False and sequence[j]==-1:
        streaking==True
        streak.append(j)
    elif streaking==True and sequence[j]==-1:
        streak.append(j)
    elif streaking==True and sequence[j]!=-1:
        streaking==False
        print(streak)
        streak=[]

print('Positive Influences:')
streaking=False
streak=[]
for j in range(len(sequence)):
    if streaking==False and sequence[j]==1:
        streaking==True
        streak.append(j)
    elif streaking==True and sequence[j]==1:
        streak.append(j)
    elif streaking==True and sequence[j]!=1:
        streaking==False
        print(streak)
        streak=[]

我希望得到以下输出：

Negative Influences: [0,1]
Positive Influences: [4,5]

不幸的是，我得到了空白：

Negative Influences:
Positive Influences:

为了进行调试，我尝试在每次迭代的末尾添加一个打印(条纹)命令。它揭示了裸奔总是错误的。为什么会这样呢？裸奔应该在达到第一个-1 (或分别为1 )时变为True。

谢谢

Answer 1

发现错误:当在条纹开始时将streaking设置为True时，我使用==而不是=。

很抱歉给您添了麻烦

Answer 2

错误地，您编写了用于将条纹值从False更改为True并将True更改为False的==，而您应该为赋值或更改变量的值编写=而不是==

错误的

streaking==True

Right

streaking=True

Answer 3

既然你知道你的代码出了什么问题，我建议用另一种简单的方法在一次迭代中完成这项工作：

from itertools import groupby

sequence = [-1,0,-1,-1,-1,-1,-1,-1,-1,-1,0,0,1,1,1,1]

pos, neg = [], []
cnt = 0

for k, g in groupby(sequence):
    g = list(g)
    if k == -1 and len(g) == 1:
        neg.append(cnt)
    elif k == -1 and len(g) > 1:
        neg.append([cnt, cnt + len(g) - 1])
    elif k == 1 and len(g) == 1:
        pos.append(cnt)
    elif k == 1 and len(g) > 1:
        pos.append([cnt, cnt + len(g) - 1])
    cnt += len(g)

print(neg)  # [0, [2, 9]]
print(pos)  # [[12, 15]]