注:每周一到周五都会进行相关Python基础知识更新,欢迎大家提宝贵的意见
一、用列表推导式取代map和filter
python 提供一种精炼的写法,可以根据一份列表来制作另外一份列表。这种表达式称为list comprehension(列表推导式)。
eg:
>>> a = [1,2,3,4,5,6]
>>> squares = [x**2 for x in a]
>>> squares
[1, 4, 9, 16, 25, 36]
除非是调用只有一个参数的函数,否则,对于简单的情况来说,列表推导式要比内置的map函数更清晰。如果使用map那么就要创建lambda函数,以便计算新列表中各个元素的值,这会是代码看起来有些乱。
使用lambda表达式形式为:
>>> squares = map(lambda x:x**2, a)
>>> squares
[1, 4, 9, 16, 25, 36]
列表推导式则不像map那么复杂,他可以直接过滤原列表中的元素,是的生成的新列表不会包含对应的计算结果。
eg:
>>> squares_2 = [x**2 for x in a if x % 2 == 0]
>>> squares_2
[4, 16, 36]
把内置的filter函数和map结合起来,也能达到同样的效果,但是代码写起来比较难懂。
eg:
latt = map(lambda x: x**2, filter(lambda x: x%2==0, a))
assert even_squares == list(latt)
总结:
1、列表推导要比内置的map和filter函数清晰,因为它唔需要额外的编写lambda表达式
2、列表推导可以跳过输入列表中的某些元素,如果改用map来做,那就必须辅以filter方能实现
3、字典和集也支持推导表达式