一个常见的任务是在一个列表上迭代,并根据条件删除一些元素。本文将展示如何完成该任务的不同方法,同时展示一些需要避免的陷阱。
假设我们需要修改列表a,并且必须删除所有不是偶数的项。首先实现辅助函数even(x)来确定一个数字x是否是偶数:
a = [1, 2, 2, 3, 4]
def even(x):
return x % 2 == 0
使用列表推导创建一个新的列表,只包含你不想删除的元素,并把它分配回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小技巧文章中了解更多关于列表推导的知识。
上面的代码创建了一个新的变量a。我们也可以通过赋值给切片a[:]就地改变现有的列表。这种方法更有效率,如果有其他对a的引用需要反映变化的话,这种方法可能很有用。
a = [1, 2, 2, 3, 4]
def even(x):
return x % 2