Python - 遍历列表时删除元素的正确做法

Python-遍历列表时删除元素的正确做法

一.问题描述

因为删除元素后，整个列表的元素会往前移动，而i却是在最初就已经确定了，是不断增大的，所以并不能得到想要的结果。
问题现象如下:

a = [1, 2, 3, 4, 5, 6, 7, 8]
for i in a:
    if i > 5:
        a.remove(i)
print(a)

运行结果：

[1, 2, 3, 4, 5, 7]

本来想要删除a中比5大的元素,但是因为6删除后,后面的元素前移了,索引变小,但是i的值却还是不断增加,就造成了这种结果

遍历列表时删除元素的正确做法如下:

• 方式一:
  遍历在新的列表操作，删除是在原来的列表操作

a = [1,2,3,4,5,6,7,8]
print(id(a))
print(id(a[:]))
for i in a[:]:
    if i>5:
    	a.remove(i)

• 方式二:列表解析

a=[1,2,3,4,5,6,7,8]
b = [i for i in a if i <= 5]
print(b) # [1, 2, 3, 4, 5]

• 方式三:倒序删除 因为列表总是‘向前移’，所以可以倒序遍历，即使后面的元素被修改了，还没有被遍历的元素和其坐标还是保持不变的

a=[1,2,3,4,5,6,7,8]
print(id(a))
for i in range(len(a)-1,-1,-1):
    if a[i]>5:
        a.remove(a[i])
print(a) # [1, 2, 3, 4, 5]