解决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]

 

posted @ 2020-12-31 13:49  致远方  阅读(1349)  评论(0编辑  收藏  举报