python列表推导的几个例子

以下内容译自python文档中关于列表推导的说明，基本上包含了常见的几种列表推导方式，这里把它们记录下来也是为了防止遗忘，也方便以后查询。

1. 平方列表

如果你想创建一个包包含1到10的平方的列表，可以用下面的方法：

>>>squares = []
>>>for x in range(10):
...squares.append(x**2)
...
>>>squares
[0,1,4,9,16,25,36,49,64,81]

但是使用列表推导可以更简洁地实现：

>>>squares = [x**2 for x in range(10)]

另外，还有一种等价的方法，要用到lambd演算：

>>>squares = map(lambd x:x**2,range(10))

相比较而言，还是上面的方法更简洁直观。

2.其它方法

>>> vec = [-4, -2, 0, 2, 4]
>>> # 原列表的所有元素乘以2，成生一个新的列表
>>> [x*2 for x in vec]
[-8, -4, 0, 4, 8]
>>> # 通过if条件筛选掉小于0的元素
>>> [x for x in vec if x >= 0]
[0, 2, 4]
>>> # 可以将函数作用于所有元素从而生成一个新的列表
>>> [abs(x) for x in vec]
[4, 2, 0, 2, 4]
>>> # 调用列表元素的对象方法来生成新列表
>>> freshfruit = ['  banana', '  loganberry ', 'passion fruit  ']
>>> [weapon.strip() for weapon in freshfruit]
['banana', 'loganberry', 'passion fruit']
>>> # 生成双元组列表，比如 (number, square)
>>> [(x, x**2) for x in range(6)]
[(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25)]
>>> # 推导式中的元组必须用括号括起来，否则会报错
>>> [x, x**2 for x in range(6)]
File "<stdin>", line 1
[x, x**2 for x in range(6)]
^
SyntaxError: invalid syntax
>>> # 列表降维，从嵌套的二维列表降为一维列表
>>> vec = [[1,2,3], [4,5,6], [7,8,9]]
>>> [num for elem in vec for num in elem]
[1, 2, 3, 4, 5, 6, 7, 8, 9]