【python】numpy list中的.remove()没有把符合条件的元素删除干净

原代码:

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

原输出:
[1, 3, 5]

很明显有问题,列表里应该只留下1。
可以看出它跳了一个来删除,原因是在遍历list时是用索引来调用的,当一个元素被remove时,下一个元素已经往前移了一个,不会被遍历到,也就无法判断条件无法删除了。
有两个解决办法:
【【python】numpy 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 deepcopylist0 = [1, 2, 3, 4, 5] list1 = deepcopy(list0)for i in list0: if i >= 2: list1.remove(i)print(list1)

[1]

    推荐阅读