日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

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

作者:程序員涵涵2021 ? 更新時間: 2022-05-18 編程語言

前言:

看似簡單的任務,往往隱藏陷阱!

一個常見的任務是在一個列表上迭代,并根據條件刪除一些元素。本文將展示如何完成該任務的不同方法,同時展示一些需要避免的陷阱。

假設我們需要修改列表a,并且必須刪除所有不是偶數的項。

首先實現輔助函數even(x)來確定一個數字x是否是偶數:

a = [1, 2, 2, 3, 4]
def even(x):
? ? return x % 2 == 0

方法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)

原文鏈接:https://blog.csdn.net/weixin_56659172/article/details/123475082

欄目分類
最近更新