前言:
看似简单的任务,往往隐藏陷阱!
一个常见的任务是在一个列表上迭代,并根据条件删除一些元素。本文将展示如何完成该任务的不同方法,同时展示一些需要避免的陷阱。
假设我们需要修改列表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)
到此这篇关于python 安全地删除列表元素的方法的文章就介绍到这了,更多相关python 删除列表元素内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!