python迭代列表中删除项目的问题

python迭代列表中删除项目的问题

问题:

有这段代码:

numbers = range(1, 50)
 
for i in numbers:
    if i < 20:
        numbers.remove(i)
 
print(numbers)

但得到的结果是:

[2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49]

当然,希望低于 20 的数字不会出现在结果中。看起来在删除时做错了什么。

解决方法

在迭代列表时正在修改列表。这意味着第一次通过循环,i == 1,所以1从列表中删除。然后for循环转到列表中的第二项,它不是2,而是 3!然后从列表中删除它,然后for循环继续到列表中的第三项,现在是 5。依此类推。也许像这样更容易形象化,用 ^ 指向的值i:

[1, 2, 3, 4, 5, 6...]
 ^

这就是列表最初的状态;然后 1 被删除,循环转到列表中的第二项:

[2, 3, 4, 5, 6...]
    ^
[2, 4, 5, 6...]
       ^

等等。

在迭代列表时没有改变列表长度的好方法。你能做的最好的事情是这样的:

numbers = [n for n in numbers if n >= 20]

或者这个,对于就地更改(括号中的东西是一个生成器表达式,它在切片赋值之前被隐式转换为一个元组):

numbers[:] = (n for in in numbers if n >= 20)

如果您想在删除之前对 n 执行操作,您可以尝试的一个技巧是:

for i, n in enumerate(numbers):
    if n < 20 :
        print("do something")
        numbers[i] = None
numbers = [n for n in numbers if n is not None]


微信扫描下方的二维码阅读更多精彩内容

python迭代列表中删除项目的问题


感谢对大飞哥的支持

相关新闻