【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]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值