python 列表推导式

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

相当于squares = map(lambda x: x**2, range(10))，但是更简洁和易读。傻逼才会用最古老的for循环方法去做这样的事情。

列表推导式由括号括起来，

括号里面包含一个表达式，

表达式后面跟着一个for语句，

后面还可以接零个或更多的for或if语句。

结果是一个新的列表，

由表达式依据其后面的for和if子句上下文计算而来的结果构成。

例如，

下面的 listcomp 组合两个列表中不相等的元素：

>>>

>>> [(x, y) for x in [1,2,3] for y in [3,1,4] if x != y]
[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

如果表达式是一个元组（例如前面示例中的 (x， y)），它必须带圆括号。

[(x, x**2) for x in range(6)]
[(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25)]

嵌套的列表推导式

考虑下面由三个长度为 4 的列表组成的 3x4 矩阵：

>>>

>>> matrix = [
...     [1, 2, 3, 4],
...     [5, 6, 7, 8],
...     [9, 10, 11, 12],
... ]

[[row[i] for row in matrix] for i in range(4)] [[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]

在实际中，与复杂的控制流比起来，

你应该更喜欢内置的函数。

针对这种场景，使用zip() 函数会更好：

>>>

>>> zip(*matrix)
[(1, 5, 9), (2, 6, 10), (3, 7, 11), (4, 8, 12)]