欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【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]
相关标签: Bug记录