【python】list中的.remove()没有把符合条件的元素删除干净
程序员文章站
2024-02-14 17:19:22
...
原代码:
list0 = [1, 2, 3, 4, 5]
for i in list0:
if i >= 2:
list0.remove(i)
print(list0)
原输出:
[1, 3, 5]
很明显有问题,列表里应该只留下1。
可以看出它跳了一个来删除,原因是在遍历list时是用索引来调用的,当一个元素被remove时,下一个元素已经往前移了一个,不会被遍历到,也就无法判断条件无法删除了。
有两个解决办法:
1、生成一个新的list,用.append()把符合条件的元素加进去(注意判断条件要取反):
list0 = [1, 2, 3, 4, 5]
list1 = []
for i in list0:
if i < 2:
list1.append(i)
print(list1)
[1]
2、用.deepcopy()复制一个原list,一个用来遍历,一个用来删除元素(注意不能用简单的copy):
from copy import deepcopy
list0 = [1, 2, 3, 4, 5]
list1 = deepcopy(list0)
for i in list0:
if i >= 2:
list1.remove(i)
print(list1)
[1]