前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >千万不要一边遍历列表一边删除元素

千万不要一边遍历列表一边删除元素

作者头像
Crossin先生
发布2023-12-19 13:21:53
1570
发布2023-12-19 13:21:53
举报

大家好,欢迎来到 Crossin的编程教室 !

今天分享一个看起来很小,但很多人都可能踩过的一个坑。

来看这个例子:

代码语言:javascript
复制
arr = [1, 2, 4, 6, 7, 8, 9]
for i in arr:
    if i % 2 == 0:
        arr.remove(i)
print(arr)

这个代码的意图很简单,就是有一个 list 列表,想删除里面的偶数值。

于是用一个 for 循环遍历列表,然后用 if 判断元素能被 2 整除的话,就从列表里删除。

但令人意外的是,结果里仍然还是有偶数。

导致这种结果的原因是因为在通过 remove 方法删除元素的时候改变了列表原本的结构:

当 i 遍历到元素 2 时,根据判断删除了这个元素,使得后面的元素索引都向前移了一位

而进入下一次遍历时,i 的值就不是下一个元素 4,而是原本下一个索引位置对应的新元素 6

这就导致元素 4 被跳过了。

所以,

不要在遍历列表的同时删除元素!

不要在遍历列表的同时删除元素!

不要在遍历列表的同时删除元素!

如果你想删,一种解决办法是手动对遍历的索引进行控制:

代码语言:javascript
复制
arr = [1, 2, 4, 6, 7, 8, 9]
i = 0
while i < len(arr):
    if arr[i] % 2 == 0:
        arr.pop(i)
    else:
        i += 1
print(arr)

但这种解法稍显复杂。所以可以转换一下思路,把不要删的元素添加到一个新的列表中:

代码语言:javascript
复制
arr = [1, 2, 4, 6, 7, 8, 9]
arr2 = []
for i in arr:
    if i % 2 != 0:
        arr2.append(i)
print(arr2)

或者直接用一个列表解析式搞定:

代码语言:javascript
复制
arr = [1, 2, 4, 6, 7, 8, 9]
arr2 = [i for i in arr if i % 2 != 0]
print(arr2)

这就方便多了。

你还有什么遍历删除列表中特定元素的方法?欢迎在评论区讨论。

感谢转发点赞的各位~


本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2023-12-18,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 Crossin的编程教室 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档