用列表的方法.remove 删除列表中元素的时候,会改变列表的下标,从而发生出乎意料的错误。所以如果想要删除列表中元素,可以用如下方法:
错误的方法:
a = [1,2,3,4,5]
b = [1,2]
for i in a:
if i in b:
a.remove(i)
# a = [2, 3, 4, 5]
思考:删除元素之后index是如何变化的。
正确的姿势:
a = [1,2,3,4,5]
b = [1,4]
c = [i for i in a if i not in b]
# c = [2,3,5]
python的list在遍历的时候删除元素一定要小心,插入也同样要小心
在迭代过程中修改迭代序列不安全(只有在使用链表这样的可变序列时才会有这样的情况)。如果你想要修改你迭代的序列(例如:复制选择项),你可以迭代它的复本。使用切割标识就可以很方便地做到这一点:
words = ['cat', 'window', 'defenestrate']
for w in words[:]: # words的浅拷贝
if len(w) > 6:
words.insert(0, w)