提问人:Finger twist 提问时间:6/7/2011 最后编辑:Super Kai - Kazuya ItoFinger twist 更新时间:7/1/2023 访问量:23212
在 Python 中迭代时从列表中删除项目时出现奇怪的结果
Strange result when removing item from a list while iterating over it in 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 的数字不会出现在结果中。看起来我在删除时做错了什么。
答:
在循环访问列表时,您正在修改列表。这意味着第一次通过循环时,so 将从列表中删除。然后循环转到列表中的第二项,它不是 ,而是 !然后从列表中删除它,然后循环继续到列表中的第三项,现在是 5。等等。也许这样更容易可视化,^ 指向 的值:i == 1
1
for
2
3
for
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 中的东西是一个生成器表达式,它在 slice-assign 之前隐式转换为元组):
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]
评论
for
"
None
None
@senderle的答案是要走的路!
话虽如此,为了进一步说明您的问题,如果您考虑一下,您将始终希望删除索引 0 20 次:
[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]
我希望它有所帮助。
下面的那些不是坏做法 AFAIK。
编辑(更多):
lis = range(50)
lis = lis[20:]
也会做这项工作。
EDIT2(我很无聊):
functional = filter(lambda x: x> 20, range(50))
从列表的末尾开始,然后向后:
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]
评论
所以我找到了一个解决方案,但它真的很笨拙......
首先,你创建一个索引数组,其中列出了所有要删除的索引,如下所示
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
它会起作用,但我想这不是预期的方法
作为 @Senderle 答案的附加信息,仅用于记录,我认为当 python 看到“序列类型”时,可视化场景背后的逻辑是有帮助的。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 上使用它时,有一种 try-catch 机制(虽然有点不同 - calling 和 Exception)。for
next()
StopIteration
*我想说的是,python 将在这里跟踪一个名为 ,因此无论列表发生什么(删除或添加),python 都会递增该变量并使用“此变量”调用方法并请求项目。index
__getitem__()
还可以使用 continue 忽略小于 20 的值
mylist = []
for i in range(51):
if i<20:
continue
else:
mylist.append(i)
print(mylist)
建立在@eyquem的答案之上并简化答案......
问题在于,当你迭代时,元素会从你下面被拉出来,当你进入下一个数字时会跳过数字。
如果您从末尾开始向后退,则随时随地删除项目将无关紧要,因为当它步入“下一个”项目(实际上是上一个项目)时,删除不会影响列表的前半部分。
只需添加到迭代器即可解决问题。注释将是一个很好的形式,可以防止未来的开发人员“整理”您的代码并神秘地破坏它。reversed()
for i in reversed(numbers): # `reversed` so removing doesn't foobar iteration
if i < 20:
numbers.remove(i)
从 Python 3.3 开始,您可以使用 list 方法作为迭代器: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]
评论
copy
您可以使用 list() for 创建一个不同的副本,如下所示: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]
对迭代进行浅拷贝,将确保迭代的列表使用不被修改。可以使用 、 或 进行浅拷贝,这将确保要删除的元素的标识相同,并减小临时列表的大小。法典:numbers
list()
copy.copy()
...
for i in list(numbers):
if i < 20:
numbers.remove(i)
...
好吧,我需要一种这种东西。
答案是使用语句。
就我而言,它真的帮助了我。所以我想把它作为答案发布。while
numbers = list(range(1, 50))
while numbers:
current_number = numbers[0]
if current_number < 20:
numbers.remove(current_number)
else:
break
numbers
>>> [20, 21, 22, 23, 24, 25, ..., 49]
我还想添加我的特殊情况 using 和 using ,但它可能超出了范围,所以我会保留它。while
for
[:]
有点晚了,但我想补充我的答案。我找到了以下相当简单的技巧
for x in list[::-1]:
if validate(x):
list.pop(x)
这样,您就可以从头开始迭代。当您删除第 n 个元素时,元素 n+1,n+2...将它们的索引减少一个,但由于您要向后移动,因此这不会完整地保留以下(或在本例中为前面)元素。
评论