[转】Python--遍历列表时删除元素的正确做法
转自:https://blog.csdn.net/cckavin/article/details/83618306
扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄
一.问题描述
这是在工作中遇到的一段代码,原理大概和下面类似(判断某一个元素是否符合要求,不符合删除该元素,最后得到符合要求的列表):
- a = [ 1, 2, 3, 4, 5, 6, 7, 8]
- for i in a:
- if i> 5:
- pass
- else:
- a.remove(i)
- print(a)
运行结果:
SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。二.问题分析
因为删除元素后,整个列表的元素会往前移动,而i却是在最初就已经确定了,是不断增大的,所以并不能得到想要的结果。
三.解决方法
1.遍历在新的列表操作,删除是在原来的列表操作
- a = [ 1, 2, 3, 4, 5, 6, 7, 8]
- print(id(a))
- print(id(a[:]))
- for i in a[:]:
- if i> 5:
- pass
- else:
- a.remove(i)
- print(a)
- print( '-------------------------')
- print(id(a))
运行结果:
2.filter
内建函数filter()官方文档参考:https://docs.python.org/3/library/functions.html#filter
- a = [ 1, 2, 3, 4, 5, 6, 7, 8]
- b = filter( lambda x: x> 5,a)
- print(list(b))
运行结果:
3.列表解析
- a = [ 1, 2, 3, 4, 5, 6, 7, 8]
- b = [i for i in a if i > 5]
- print(b)
运行结果:
4.倒序删除
因为列表总是“向前移”,所以可以倒序遍历,即使后面的元素被修改了,还没有被遍历的元素和其坐标还是保持不变的。
- 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:
- pass
- else:
- a.remove(a[i])
- print(id(a))
- print( '--------------------')
- print(a)
运行结果:

更多精彩