在 Python 中迭代时从列表中删除项目时出现奇怪的结果

问题描述 投票:0回答:9

我有这段代码:

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 的数字。看起来我在删除时做错了。

python list loops for-loop range
9个回答
144
投票

您在迭代列表时正在修改它。这意味着第一次通过循环,

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]

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

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]

18
投票

从列表的末尾开始,然后倒退:

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]

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]

希望对您有所帮助。


以下是 not 不好的做法 AFAIK.

编辑(更多):

lis = range(50)
lis = lis[20:]

也会做这份工作。

EDIT2(我很无聊):

functional = filter(lambda x: x> 20, range(50))

3
投票

所以我找到了一个解决方案,但它真的很笨拙......

首先创建一个索引数组,在其中列出所有要删除的索引,如下所示

numbers = range(1, 50)
index_arr = []

for i in range(len(numbers):
    if numbers[i] < 20:
        index_arr.append(i)

之后,您要删除编号列表中的所有条目,索引保存在 index_arr 中。你会遇到的问题和以前一样。因此,在刚刚从数字 arr 中删除一个数字后,您必须从 index_arr 中的每个索引中减去 1,如下所示:

numbers = range(1, 50)
index_arr = []

for i in range(len(numbers):
    if numbers[i] < 20:
        index_arr.append(i)

for del_index in index_list:
    numbers.pop(del_index)

    #the nasty part
    for i in range(len(index_list)):
        index_list[i] -= 1

它会起作用,但我想这不是预期的方法


1
投票

作为@Senderle 答案的附加信息,仅供记录,我认为当 python 在“

Sequence type
”上看到for 时,可视化场景背后的逻辑是有帮助的。

假设我们有:

lst = [1, 2, 3, 4, 5]

for i in lst:
    print(i ** 2)

实际上会是:

index = 0
while True:
    try:
        i = lst.__getitem__(index)
    except IndexError:
        break
    print(i ** 2)
    index += 1

就是这样,当我们在 Sequence 类型或 Iterables 上使用它时,

for
有一个 try-catch 机制(虽然有点不同 - 调用
next()
StopIteration
异常)。

*我想说的是,python 将在这里跟踪一个名为

index
的自变量,因此无论列表发生什么(删除或添加),python 都会递增该变量并调用
__getitem__()
方法“这个变量”并要求项目。


1
投票

建立并简化@eyquem的答案...

问题在于,当您迭代时,元素正在从您的下方被拉出,在您前进到 was 下一个数字时跳过数字。

如果你从末尾开始往回走,在移动中删除项目将无所谓,因为当它走到“下一个”项目(实际上是前一个项目)时,删除不会影响前半部分列表。

只需将

reversed()
添加到迭代器即可解决问题。评论是一种很好的形式,可以防止未来的开发人员“整理”您的代码并神秘地破坏它。

for i in reversed(numbers): # `reversed` so removing doesn't foobar iteration
  if i < 20:
    numbers.remove(i)

0
投票

您也可以使用 continue 忽略小于 20 的值

mylist = []

for i in range(51):
    if i<20:
        continue
    else:
        mylist.append(i)
print(mylist)

0
投票

Python 3.3 起,您可以使用列表

copy()
方法作为迭代器:

numbers = list(range(1, 50))

for i in numbers.copy():
    if i < 20:
        numbers.remove(i)
print(numbers)

[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]

0
投票

您可以使用 list()

numbers
创建不同的副本
numbers
如下所示:

numbers = list(range(1, 50))
       # ↓ ↓ Here ↓ ↓
for i in list(numbers):
    if i < 20:
        numbers.remove(i)

print(numbers) # [20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 
               #  31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 
               #  f42, 43, 44, 45, 46, 47, 48, 49]
© www.soinside.com 2019 - 2024. All rights reserved.