python 安全地刪除列表元素的方法

2022-03-15 19:00:28

方法1: 建立新列表，過濾元素

1a) 列表推導，建立新列表

使用列表推導建立一個新的列表，只包含你不想刪除的元素，並把它分配回a：

a = [1, 2, 2, 3, 4]
def even(x):
    return x % 2 == 0
 
# 列表推導，但建立了一個新的變數a
a = [x for x in a if not even(x)]
# --> a = [1, 3]
print(a)

1b) 列表推導，對a[:]賦值

上面的程式碼建立了一個新的變數a。我們也可以通過賦值給切片a[:]就地改變現有的列表。這種方法更有效率，如果有其他對a的參照需要反映變化的話，這種方法可能很有用。

a = [1, 2, 2, 3, 4]
def even(x):
    return x % 2 == 0
 
# 列表推導，但賦值給a[:] 就地改變列表
a[:] = [x for x in a if not even(x)]
# --> a = [1, 3]
print(a)

1c) 使用itertools.filterfalse()

itertools模組為非常有效的迴圈迭代提供了各種函數，並且提供了一種過濾元素的方法。

a = [1, 2, 2, 3, 4]
def even(x):
    return x % 2 == 0
 
# 通過itertools 快速過濾
from itertools import filterfalse
a[:] = filterfalse(even, a)
# --> a = [1, 3]
print(a)

方法2：列表副本上迭代

如果你真的想保留for語法，那麼需要在列表的副本上進行迭代（副本可以通過使用a[:]簡單建立）。

現在你可以在條件為True時從原始列表中刪除元素:

a = [1, 2, 2, 3, 4]
def even(x):
    return x % 2 == 0
 
# 注意是在列表副本a[:] 上回圈
for item in a[:]:
    if even(item):
        a.remove(item)
# --> a = [1, 3]
print(a)

常見陷阱

千萬別在同一個列表上回圈，並在迭代過程中修改它!

這和上面的程式碼是一樣的，只是沒有在副本上回圈。刪除一個元素將使所有後續元素向左移動一個位置，因此在下一次迭代中，一個元素將被跳過。

這可能會導致不正確的結果:

a = [1, 2, 2, 3, 4]
def even(x):
    return x % 2 == 0
 
# 直接在變數a上進行迴圈，沒有在副本上
for item in a:
    if even(item):
        a.remove(item)
# --> a = [1, 2, 3] !!!
print(a)

另外，在列表的迴圈過程中，千萬不要修改索引!

這是不正確的，因為在迴圈中改變i不會影響下一次迭代中i的值。這個例子也會產生非預期的效果，甚至會導致IndexErrors，

比如這裡：

a = [1, 2, 2, 3, 4]
def even(x):
    return x % 2 == 0
 
# 試圖在迴圈在改變索引i，但出錯！
for i in range(len(a)):
    if even(a[i]):
        del a[i]
        i -= 1
# --> IndexError: list index out of range
print(a)

到此這篇關於python 安全地刪除列表元素的方法的文章就介紹到這了,更多相關python 刪除列表元素內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！