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

假設(shè)我們需要修改列表a，并且必須刪除所有不是偶數(shù)的項(xiàng)。首先實(shí)現(xiàn)輔助函數(shù)even(x)來(lái)確定一個(gè)數(shù)字x是否是偶數(shù):

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

方法1: 創(chuàng)建新列表，過(guò)濾元素

1a) 列表推導(dǎo)，創(chuàng)建新列表

使用列表推導(dǎo)創(chuàng)建一個(gè)新的列表，只包含你不想刪除的元素，并把它分配回a：

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

# 列表推導(dǎo)，但創(chuàng)建了一個(gè)新的變量a
a = [x for x in a if not even(x)]
# --> a = [1, 3]
print(a)

你可以在10個(gè)python小技巧文章中了解更多關(guān)于列表推導(dǎo)的知識(shí)。

1b) 列表推導(dǎo)，對(duì)a[:]賦值

上面的代碼創(chuàng)建了一個(gè)新的變量a。我們也可以通過(guò)賦值給切片a[:]就地改變現(xiàn)有的列表。這種方法更有效率，如果有其他對(duì)a的引用需要反映變化的話，這種方法可能很有用。

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

# 列表推導(dǎo)，但賦值給a[:] 就地改變列表
a[:] = [x for x in a if not even(x)]
# --> a = [1, 3]
print(a)

1c) 使用itertools.filterfalse()

itertools模塊為非常有效的循環(huán)迭代提供了各種函數(shù)，并且提供了一種過(guò)濾元素的方法。

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

# 通過(guò)itertools 快速過(guò)濾
from itertools import filterfalse
a[:] = filterfalse(even, a)
# --> a = [1, 3]
print(a)

方法2：列表副本上迭代

如果你真的想保留for語(yǔ)法，那么需要在列表的副本上進(jìn)行迭代（副本可以通過(guò)使用a[:]簡(jiǎn)單創(chuàng)建）?，F(xiàn)在你可以在條件為True時(shí)從原始列表中刪除元素:

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

# 注意是在列表副本a[:] 上循環(huán)
for item in a[:]:
    if even(item):
        a.remove(item)
# --> a = [1, 3]
print(a)

常見陷阱

千萬(wàn)別在同一個(gè)列表上循環(huán)，并在迭代過(guò)程中修改它!

這和上面的代碼是一樣的，只是沒(méi)有在副本上循環(huán)。刪除一個(gè)元素將使所有后續(xù)元素向左移動(dòng)一個(gè)位置，因此在下一次迭代中，一個(gè)元素將被跳過(guò)。這可能會(huì)導(dǎo)致不正確的結(jié)果:

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

# 直接在變量a上進(jìn)行循環(huán)，沒(méi)有在副本上
for item in a:
    if even(item):
        a.remove(item)
# --> a = [1, 2, 3] !!!
print(a)

另外，在列表的循環(huán)過(guò)程中，千萬(wàn)不要修改索引!

這是不正確的，因?yàn)樵谘h(huán)中改變i不會(huì)影響下一次迭代中i的值。這個(gè)例子也會(huì)產(chǎn)生非預(yù)期的效果，甚至?xí)?dǎo)致IndexErrors，比如這里：

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

# 試圖在循環(huán)在改變索引i，但出錯(cuò)！
for i in range(len(a)):
    if even(a[i]):
        del a[i]
        i -= 1
# --> IndexError: list index out of range
print(a)

小節(jié)

通過(guò)上面的學(xué)習(xí)相信你現(xiàn)在可以安心地刪除列表的部分元素.