最近在看别人写的python程序时,经常看到下面的一种写法,让我觉得非常的简洁:
一开始并不知道这是什么样的高级语法(用phthon不是很多),好像在网上简单的搜索并学习了一个,这种格式在python的官方名称是List Comprehensions, 度娘叫:列表推导式。print [x for x in list]
List Comprehensions主要是可以列表的遍历或以更加方便,简洁。 举个例子:
def power(list):
powered_list = []
for x in list:
powered_list.append(x*x)
return powered_list
上面这个函数对一个数组内的元素进行乘方运算,然后返回一个新的数组。可以看到这里至少要用4行的代码。
那如果我们换成List Comprehension的写法,如下:
可以看到我们只用一行代码就可以完成对数组的迭代以及运算。是不是感觉简单,方便很多。def power(list):
return [x*x for x in list]
再回到上面的例子,如果我们把程序改得稍微复杂一点,比较只对list里面是2的整数倍的做乘方运算,并把乘方结果放到新的list里面,然后返回这个新list。
我们来看一下用传统的写法是怎么样的:
def power(list):
powered_list = []
for x in list:
if x%2 == 0:
powered_list.append(x*x)
return powered_list
然后再对比一个用List Comprehensions后的代码:
没有对比就没有伤害,可以看得出用List Comprehension在代码简洁上明显好很多。def power(list)
return [x*x for x in list if x%2 == 0]
总结List Comprehension的基本语法结构就是:
[x for x in list if is ...]
[(x,y) for x in list1 for y in list2]
List Comprehension的更详细的使用可以参考
PEP 202 (Python Enhancement Proposal)