前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Python遍历列表remove漏删解决方案

Python遍历列表remove漏删解决方案

作者头像
梦无矶小仔
发布2022-12-20 16:37:17
5160
发布2022-12-20 16:37:17
举报

python系列

遍历列表remove漏删解决方案

需求及背景

我有一个需求:保留list_1中含有list_2元素的元素,其他的删除

代码语言:javascript
复制
list_1 = ['魁哥','夭夭', '阿离', '渣姐', '秃头企鹅', '王哥', '小白', '卷魔']
list_2 = ['旺仔', '秃头企鹅', '王哥']

最后需要输出

代码语言:javascript
复制
['秃头企鹅', '王哥']

直接一顿操作,写出了如下代码

代码语言:javascript
复制
for i in list_1:
    if i not in list_2:
        list_1.remove(i)
print(list_1)

结果输出的结果是

代码语言:javascript
复制
['夭夭', '渣姐', '秃头企鹅', '王哥', '卷魔']

为什么会这样?通过debug发现,每删除一个元素,list_1里面的元素下标就会发生变化,而for循环里面是持续性的读取下一个值的,每次删除一个的时候,便会跳过一个。

既然如此我就是用下标删呗

代码语言:javascript
复制
for i in range(len(list_1)):
if list_1[i] not in list_2:
        list_1.remove(list_1[i])
continue
print(list_1)

这时候又发现,直接报错

代码语言:javascript
复制
IndexError: list index out of range

当删除一个元素之后,新列表的长度减少,i还是根据原来列表的索引取值,当i超过当前的列表长度时,新列表没有该下标对应元素,索引越界。

解决方案

方案一(快捷)

代码语言:javascript
复制
list_1 = ['夭夭', '阿离', '渣姐', '秃头企鹅', '魁哥', '小白', '卷魔']
list_2 = ['旺仔', '秃头企鹅', '王哥']
j = 0
for i in range(len(list_1)):
    if list_1[j] not in list_2:
        list_1.remove(list_1[j])
        continue
    j+=1
print(list_1)

方案一方法的思路:

每次只要删除了,都从上一次的下标继续,如果这次没删除,则下标加一删下一个。看不太懂的可以丢自己编辑器里面加一些print进行分步分析。

输出

代码语言:javascript
复制
['秃头企鹅', '王哥']

方案二(简单)

倒序 如果让索引倒序遍历列表就不会出现越界的问题了。

代码语言:javascript
复制
for i in range(len(list_1)-1,-1,-1):
    if list_1[i] not in list_2:
        list_1.remove(list_1[i])
print(list_1)

方案二方法思路:

将索引倒序,这样不管你怎么删,我下标都是倒着来,肯定不会下标越界。(此方法对应解决最开始的下标越界的思路。)

注意细节:(自己多尝试,多写)

①len减一是因为左闭右开;

②第三个参数表示的是对数值所进行的操作,每次加上-1;

③第二个值表示最终加到-1,但不包含-1。(最终加到几,但是不包含本身)

输出

代码语言:javascript
复制
['秃头企鹅', '王哥']

方案三(粗暴)

代码语言:javascript
复制
while True:
    for i in list_1:
        if i not in list_2:
            list_1.remove(i)
            break
    else:
        break
print(list_1)

方案三思路:

无限循环,知道你搞完,这个不多解释,就是莽夫。

输出

代码语言:javascript
复制
['秃头企鹅', '王哥']

方案四(灵活)

代码语言:javascript
复制
d = dict(zip(range(len(list_1)),list_1))
for k,v in d.items():
    if v.lower() not in list_2:
        d[k]=[]
final_1 =  [v for k,v in d.items() if v]
print(final_1)

方案四思路:

讲列表转换成字典,再通过字典的value和list_2进行对比,如果不是list_2中的元素则赋值为空列表,最后拿到不是空列表的value,就是我们需要的列表了。

听起来会有点绕,自己尝试一下,一下就练了字典、列表、遍历、推导式、列表合并字典这些知识点哦。

输出

代码语言:javascript
复制
['秃头企鹅', '王哥']

上面用例列表推导式,zip,dict等方法,这些python基础我之前文章有写,大家可以多翻翻。(如果没有,那肯定是我不小心删了。)

最后,祝大家周末愉快!

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

本文分享自 梦无矶的测试开发之路 微信公众号,前往查看

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

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

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