Python中List推导式的使用方法

在Python中要创建一个包含0-10的平方值的List，一般使用如下代码

squares = []
for x in range(11):
  squares.append(x**2)

其中，squares中包含了0-10的平方值。可以使用List推导式（List Comprehension）可以用来很简便地创建上述List。

1 List推导式的格式

List推导式的格式的格式如下所示。

[(表达式) for语句 if语句]

其中，List推导式可以包含if语句，也可以不包含if语句。

2 不包含if语句的List推导式

使用如下代码可以创建一个包含0-10的平方值的List。

squares1 = [(a**2) for a in range(11)]

以上代码使用了List推导式，该推导式中的表达式是“a**2”，for语句遍历了从0-10的数字。

需要注意的是：使用文章起始处的代码创建squares的同时，还创建了一个x的变量，当代码结束后，x的值是10；而使用List推导式创建的squares1时用到的变量a，是一个局部变量，当代码结束后，变量a会被释放，如图1所示。

图1 变量a被释放

3包含if语句的List推导式

还可以在List推导式中包含if语句，代码如下所示。

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

其中，x的取值范围是[1,2,3]；y的取值范围是[3,1,4]，if语句的含义是当x不等于y时，取(x,y)的值。即通过次List推导式创建的List为

[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]