Python陷阱-如何安全地删除列表元素?

╰半夏微凉° 2024-02-05 23:22 159阅读 0赞

一个常见的任务是在一个列表上迭代，并根据条件删除一些元素。本文将展示如何完成该任务的不同方法，同时展示一些需要避免的陷阱。

假设我们需要修改列表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)

你可以在10个python小技巧文章中了解更多关于列表推导的知识。

##### **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)

#### **小结** ####

通过上面的学习相信你现在可以安心地删除列表的部分元素.