首頁 > 軟體

Python列表中多元素刪除(移除)的實現

2023-03-11 06:01:11

問題背景

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.
#(無返回值)
#輸入為具體的元素,重複的元素首先移除第一個

若直接使用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) 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列表中多元素刪除(移除)的實現的文章就介紹到這了,更多相關Python列表多元素刪除內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com