Python遍历列表时删除元素

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

出错

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

输出:

1
[2, 3, 4, 1]

另一种:

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

输出:

1
2
3
4
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()函数过滤掉符合指定条件的元素。

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

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

  2. 使用列表解析式。

    1
    2
    s4 = [i for i in s if i != 1]
    print(s4)
  3. 把原始列表拷贝给一个新列表,遍历新列表,修改原始列表(或相反)。

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

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

    1
    2
    3
    4
    5
    s5 = s
    for i in s:
    if i == 1:
    s5.remove(i)
    print(s5)

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

    1
    2
    3
    4
    5
    6
    7
    import copy

    s5 = copy.deepcopy(s)
    for i in s:
    if i == 1:
    s5.remove(i)
    print(s5)
  4. while循环来做判断条件。

    1
    2
    3
    4
    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