在Python中迭代列表时移除项的奇怪结果
我有一段代码:
numbers = list(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以下的数字。看起来我在使用remove的时候做错了什么。
12 个回答
11
@senderle的回答是个好主意!
说到这里,为了更清楚地说明你的问题,如果你仔细想想,你会发现你总是想要把第0个索引的元素删除二十次:
[1,2,3,4,5............50]
^
[2,3,4,5............50]
^
[3,4,5............50]
^
所以你其实可以用这样的方式:
aList = list(range(50))
i = 0
while i < 20:
aList.pop(0)
i += 1
print(aList) #[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]
希望这对你有帮助。
下面的做法我认为并不是坏习惯。
编辑(补充一些):
lis = range(50)
lis = lis[20:]
这也能完成任务。
编辑2(我有点无聊):
functional = filter(lambda x: x> 20, range(50))
24
从列表的末尾开始,往回走:
li = list(range(1, 15))
print(li)
for i in range(len(li) - 1, -1, -1):
if li[i] < 6:
del li[i]
print(li)
结果:
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
[6, 7, 8, 9, 10, 11, 12, 13, 14]
152
你在遍历列表的时候同时修改它。这意味着在第一次循环时,i == 1
,所以1
被从列表中移除了。然后for
循环接着去看列表中的第二个项目,但它不是2
,而是3
!接着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 n 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]