Python循环中删除列表元素会出现的’坑’
前言:在Python中,如果在循环过程中remove()删除列表元素可能会出现的一些不易发现的bug。
目的:
a=[11,22,33,44,55,66,77]
#利用for循环删除a中的33和44
错误示例:
for i in a:
if i==33 or i==44:
a.remove(i)
#输出结果:a=[11,22,44,55,66,77]
#并没有删除44,因为在删除33之后,44填补了33的位置成为了a[2],(就像排队一样,如果有
#人离开,后面的人就会自动往前走),而此时在for循环中i的值已经变为了a[3],因此,变为
#a[2]的44“逃过了一劫”
正确示例:
#为了尽量避免上面这种错误的发生,我们可以将要删
#除的元素在循环外删除,先将其存在另一个列表中。
b=[]
for i in a:
if i==33 or i==44:
b.append(i)
for i in b:
a.remove(i)
#输出结果:b=[33,44]
a=[11,22,55,66,77]