python 根据条件重置列表的枚举索引

Question

我想遍历 list1 并将递增的值附加到 list2，直到它达到 list1 中的值，其中 value != 0 然后递增的值重置回 0并再次从 0 开始枚举。

我尝试了以下方法:

list1 = [0,0,0,25,0,0,0,20,0,0]
list2 = []

for i,v in enumerate(list1):
    if v == 0:
        list2.append(i+1)
    else:
        list2.append(0)

但这会返回:

list2 = [1,2,3,0,5,6,7,0,9,10]

期望的输出:

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

Answer 1

for 和 enumerate(..) 的问题在于 for 不关心像 这样的变量设置循环中的 i:如果开始新的迭代，则会从提供给 for 的迭代器中获取 i。所以即使你设置了i，它对下一次迭代也没有影响。

因此，诀窍是在循环之外定义 i 并自己更新它(增加它，或者在 v != 0 时重置它):

list1 = [0,0,0,25,0,0,0,20,0,0]
list2 = []

<b>i = 0</b>
for <b>v</b> in <b>list1</b>:
    <b>i += 1</b>
    <b>if v != 0:</b>
        <b>i = 0</b>
    <b>list2.append(i)</b>