Python遍历列表时删除元素

本文首发于我的个人博客Suixin’s Blog
原文: https://suixinblog.cn/2019/03/remove-for-list.html　　作者: Suixin

Python的for循环可以遍历一个列表，但若需要同时将符合条件的元素删除时，则会出错。

出错

s = [1, 2, 3, 4, 1, 1]
s1 = s
for i in s1:
    if i == 1:
        s1.remove(i)
print(s1)

输出：

[2, 3, 4, 1]

另一种：

s2 = s
for idx in range(len(s2)):
    if s2[idx] == 1:
        del s2[idx]
print(s2)

输出：

Traceback (most recent call last):
  File "temp.py", line 11, in <module>
    if s2[idx] == 1:
IndexError: list index out of range

出错原因

Python中用for发起任何形式的遍历时，它的遍历顺序都是从最初就确定的，而在遍历中删除了元素会导致当前索引的变化，所以遍历中间取到的索引的那个值已经变了，会漏掉一些元素。另一种错误是删除元素后列表长度缩小了，所以会超出索引范围。

正确删除法

1. 使用filter()函数过滤掉符合指定条件的元素。

   s3 = s
   print(list(filter(lambda x: x != 1, s3)))
   

   此处lambda为Python中函数的简化定义形式。

2. 使用列表解析式。

   s4 = [i for i in s if i != 1]
   print(s4)
   

3. 把原始列表拷贝给一个新列表，遍历新列表，修改原始列表（或相反）。

   s6 = s
   for i in s6[:]:
       if i == 1:
           s6.remove(i)
   print(s6)
   

   但是，下面的赋值操作给新列表是不行的，因为新变量和原变量的物理地址是相同的，可通过id()函数查看。

   s5 = s
   for i in s:
       if i == 1:
           s5.remove(i)
   print(s5)
   

   可通过深拷贝解决上述问题：

   import copy
   
   s5 = copy.deepcopy(s)
   for i in s:
       if i == 1:
           s5.remove(i)
   print(s5)
   

4. 用while循环来做判断条件。

   s7 = s
   while 1 in s7:
       s7.remove(1)
   print(s7)
   

上述4种方法的输出皆为：[2, 3, 4]。

参考

https://segmentfault.com/a/1190000007214571
http://www.runoob.com/w3cnote/python-understanding-dict-copy-shallow-or-deep.html