【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时,下一个元素已经往前移了一个,不会被遍历到,也就无法判断条件无法删除了。

有两个解决办法:

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记录)