python 列表推导式

2023-06-08,,

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

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

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

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

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

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

结果是一个新的列表,

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

例如,

下面的 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)]
 

python 列表推导式的相关教程结束。

《python 列表推导式.doc》

下载本文的Word格式文档,以方便收藏与打印。