python - 数组中连续子序列的总和

Question

给定以下数组:

tab = [80,12,14,5,70,9,26,30,8,12,16,15]

我想计算大小为 4 的所有可能序列的总和，如下所示:

S1=80+12+14+5=111
S2=12+14+5+70 =101
S3=14+5+70+9 =98
....

我已经在 python 上实现了一个简短的程序来执行此操作，但效率不高:

import numpy as np


tab= np.array([80,12,14,5,70,9,26,30,8,12,16,15])
tab_size=tab.size
n=tab_size
s=0
seq_len=5
for i in range (0,n-(seq_len-1),1):
    print("index i ",i)
    for k in range(i+1,(seq_len+i),1):
            print ("index k ", k)
            tab[i]=tab[i]+tab[k]
            s=s+1
print(s)         
tab

结果如下:

array([111, 101,  98, 110, 135,  73,  76,  66,  51,  12,  16,  15])

我注意到每个元素将参与求和运算 4 次，这并不好。你有什么有效的想法吗？
我想补充一点，序列大小不固定，在这个例子中只是 4。提前谢谢你

Answer 1

• 一旦计算出S1，您只需加 70 减去 80 即可得到 S2。
• 计算出S2后，只需加9减12即可得到S3。
• ...

这样，您将避免使用每个元素 4 次。