解决python 循环遍历列表删除多元素及重复元素


在遍历list时,删除符合条件的数据,结果不符合预期。

bb列表中为要删除的元素值,aa为原生列表

aa = [1,2,2,3,4,5,6,7,7,8,9]
bb = [2,7]
for i in aa:
    if i in bb:
        idx = aa.index(i)
        aa.pop(i)
print(aa)

循环遍历后的结果为

[1, 2, 3, 4, 5, 6, 7, 8, 9]

原因是,删除aa中的元素后,aa的实际长度变小了,但是循环次数没有减少,依然按照原来list_aa的长度进行遍历,所以会造成索引溢出。

解决办法:

把列表拷贝,然后对原列表进行删除操作就没问题了

aa = [1,2,2,3,4,5,6,7,7,8,9]
bb = [2,7]
for i in aa[:]:
    if i in bb:
        idx = aa.index(i)
        aa.pop(idx)
print(aa)

结果为:

[1, 3, 4, 5, 6, 8, 9]

aa[:]是对原始的aa的一个拷贝,是一个新的list_aa,所以,我们遍历新的list_aa,而删除原始的list_aa中的元素,则既不会引起索引溢出,最后又能够得到想要的最终结果。此方法的缺点可能是,对于过大的list,拷贝后可能很占内存

方法二:可以倒叙遍历列表,删除

aa = [1,2,2,3,4,5,6,7,7,8,9]
bb = [2,7]
for i in aa[::-1]:
    if i in bb:
        idx = aa.index(i)
        aa.pop(idx)
print(aa)

结果为:

[1, 3, 4, 5, 6, 8, 9]

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM