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