我有一个(长)列表,其中0和1随机出现:

list_a = [1, 1, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1]

我想得到list_b

• 列表的总和直到0出现的位置

• 如果出现0,则在列表中保留0

  list_b = [1, 2, 3, 0, 1, 2, 0, 1, 0, 1, 2, 3]
  

  Run Code Online (Sandbox Code Playgroud)

我可以按如下方式实现:

list_b = []
for i, x in enumerate(list_a):
    if x == 0:
        list_b.append(x)
    else:
        sum_value = 0
        for j in list_a[i::-1]:
            if j != 0:
                sum_value += j
            else:
                break
        list_b.append(sum_value)
print(list_b)

但实际列表的长度很长.

所以,我想改进高速代码.(如果不可读)

我改变了这样的代码:

from itertools import takewhile
list_c = [sum(takewhile(lambda x: x != 0, list_a[i::-1])) …

lsit_a = [2, 4, 3, 6, 3, 8, 5]

列表理解非常有用。

list_b = [a**2 for a in list_a]

我想知道如何在python列表理解中编写自引用。

例如 -

list_c = [a**2 if i == 0 else a*2 + (itself[i-1]) for i, a in enumurate(list_a)]

怎么写的一部分itself[i-1]？

加

对不起，我的解释不好。

我想通过高速方式获得这份清单。

输入清单

list_a = [0,0,1,0,0,0,1,0,0,1,0]

输出清单

list_c = [0，0，1，0.9，0.8，0.7，1，0.9，0.8，1，0.9]

就像锯齿列表。