Python 列表中多元素的删除(移除)


 


问题背景

a=[1,4,2,1,5,6,9,0] #删除列表中的元素,其所在的位置为[1,3,7] del_index=[1,3,7] tmp=[i for num,i in enumerate(a) if num not in del_index] print(f"最终结果:{tmp}") 

结果:

最终结果:[1, 2, 5, 6, 9] 

目的是实现List中的多个元素的移除(单个元素的比较简单,这里不做过多赘述)。而多元素的删除,可以归纳为以下两类:

实现方法

1. 使用枚举法

直接使用枚举,然后判断索引在不在删除索引里面即可。

[i for num,i in enumerate(a) if num not in del_index] 
2. python中List的内置方法

使用内置方法remove移除或pop,但是每次只能移除一个。具体的函数差异:

#pop L.pop([index]) -> item -- remove and return item at index (default last).#(从后往前,有返回值) #输入为:元素所在的索引 #remove: L.remove(value) -> None -- remove first occurrence of value. #(无返回值) #输入为具体的元素,重复的元素首先移除第一个 
  1. 若直接使用for循环直接使用上述函数,就会报错!!!比如:


--------------------------------------------------------
IndexError             Traceback (most recent call last)
<ipython-input-25-364cdd187e0c> in <module> 4 b=a.copy() 5 for i in del_index: ----> 6 b.remove(b[i])# b.pop(i) 7 print(b) 8 IndexError: list index out of range 
  1. 改进:

(1) pop


b=a.copy()
for num,i in enumerate(del_index): print(f"{b},被移除的元素:{a[i]}") b.pop(i-num) print(f"最终结果:{b}") 

结果与枚举法结果一致。
(2) remove

c=a.copy()
for num,i in enumerate(del_index): print(f"{c},被移除的元素:{a[i]}") c.remove(c[i-num])#重复元素 print(f"最终结果:{c}") 

输出结果:


[1, 4, 2, 1, 5, 6, 9, 0],被移除的元素:4 [1, 2, 1, 5, 6, 9, 0],被移除的元素:1 [2, 1, 5, 6, 9, 0],被移除的元素:0 最终结果:[2, 1, 5, 6, 9] 

remove后的结果与上述方法的结果不一致.主要原因是重复值的问题,a[3]=1,a[0]=1,所以在移除元素4后,原列表变成[1, 2, 1, 5, 6, 9, 0].移除元素1时,本应该移除的是4右侧的1,但是由于remove的默认属性,它会选择第一次出现的1进行移除。虽然得到的结果值一样,但是元素的索引与其它方法的结果不一致。

结论

就最终结果而言,枚举法pop法不受重复值的影响;而remove容易受重复值的影响。

若最终结果不受顺序影响,三者的差异可以忽略不计。

 
分类:  Python


免责声明!

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



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