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

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

@[toc]

需求及背景

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

list_1 = ['魁哥','夭夭', '阿离', '渣姐', '秃头企鹅', '王哥', '小白', '卷魔']
list_2 = ['旺仔', '秃头企鹅', '王哥']

最后需要输出

['秃头企鹅', '王哥']

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

for i in list_1:
    if i not in list_2:
        list_1.remove(i)
print(list_1)

结果输出的结果是

['夭夭', '渣姐', '秃头企鹅', '王哥', '卷魔']

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

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

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)

这时候又发现,直接报错

IndexError: list index out of range

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

解决办法

方案一(快捷)

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进行分步分析。

输出

['秃头企鹅', '王哥']

方案二(简单)

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

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。(最终加到几,但是不包含本身)

输出

['秃头企鹅', '王哥']

方案三(粗暴)

while True:
    for i in list_1:
        if i not in list_2:
            list_1.remove(i)
            break
    else:
        break
print(list_1)

方案三思路:

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

输出

['秃头企鹅', '王哥']

方案四(灵活)

d = dict(zip(range(len(list_1)),list_1))
for k,v in d.items():
    if v 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,就是我们需要的列表了。

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

输出

['秃头企鹅', '王哥']

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

最后,祝大家周末愉快!

gzh:梦无矶的测试开发之路

相关文章
|
4天前
|
索引 Python
Python 中寻找列表最大值位置的方法
本文介绍了Python中找列表最大值及其位置的三种方法:1) 使用内置`max()`和`index()`函数;2) 通过循环遍历;3) 利用`enumerate()`函数和生成器表达式。每种方法均附有示例代码,其中`enumerate()`方法在保证效率的同时代码更简洁。
25 2
|
4天前
|
存储 运维 数据挖掘
Python列表中每个元素前面连续重复次数的数列统计
Python列表中每个元素前面连续重复次数的数列统计
12 1
|
4天前
|
存储 JSON 数据库
Python中列表数据的保存与读取:以txt文件为例
Python中列表数据的保存与读取:以txt文件为例
18 2
|
2天前
|
数据采集 数据挖掘 Python
10个python小技巧,优雅地书写人生_python列表遍历奇数偶数
10个python小技巧,优雅地书写人生_python列表遍历奇数偶数
|
2天前
|
存储 索引 Python
【python学习】列表、元组、字典、集合,秋招是不是得到处面试
【python学习】列表、元组、字典、集合,秋招是不是得到处面试
|
4天前
|
存储 机器学习/深度学习 数据可视化
基于Python的数据分组技术:将数据按照1, 2, 3规则分为三个列表
基于Python的数据分组技术:将数据按照1, 2, 3规则分为三个列表
9 1
|
4天前
|
数据挖掘 计算机视觉 Python
Python实现对规整的二维列表中每个子列表对应的值求和
Python实现对规整的二维列表中每个子列表对应的值求和
11 0
|
4天前
|
存储 数据采集 数据可视化
Python列表到Excel表格第一列的转换技术详解
Python列表到Excel表格第一列的转换技术详解
9 0
|
4天前
|
Python
【Python操作基础】——列表操作
【Python操作基础】——列表操作
|
4天前
|
索引 Python
Python中的列表、元组和字典各具特色
【5月更文挑战第11天】Python中的列表、元组和字典各具特色:列表是可变的,元组不可变,字典亦可变;列表和元组有序,字典无序(但在Python 3.7+保持插入顺序);元素类型上,列表和元组元素任意,字典需键不可变;列表用方括号[],元组用圆括号(),字典用大括号{}表示。列表不适合作字典键,元组可以。选择数据结构应依据实际需求。
22 2