列表推导式 - 码农教程

学了python很长时间，我才真正的理解列表推导式。

基本语法

a = [i*i for i in range(3)]
print(a)

结果：

[0,1,4]

基本原理：Python在执行列表推导式时，会对可迭代对象 L 进行迭代，将每一次迭代的值赋给循环变量 i ，然后收集变量表达式 i*i 的计算结果，最终由这些结果构成了新的列表，也就是列表推导式所返回的值。

只要支持 for 循环进行迭代的对象，都可以对它使用列表推导式。

高级用法

带if语句的

我们可以在 for 语句后面跟上一个 if 判断语句，用于过滤掉那些不满足条件的结果项。

例如，一行打印出1~100偶数的列表可以这么写

a = [ i for i in range (1,100) if i%2==0]
print(a)

带有嵌套循环的

在复杂一点的列表推导式中，可以嵌套有多个 for 语句。按照从左至右的顺序，分别是外层循环到内层循环。

b = [x+y for y in ["A","B"] for x in ["1","2"]]
print(b)

结果：

['1A', '2A', '1B', '2B']

相当于：

li = []
for x in ["1","2"]:
    for y in ["A","B"]:
        li.append(x+y)
print(li)

既有if语句又有for嵌套循环

例题：输出0~4之间偶数和奇数的和：

b = [(i,j)for j in range(5)if j %2==0 for i in range(5) if i%2==0]
print(b)

结果：

[(0, 0), (2, 0), (4, 0), (0, 2), (2, 2), (4, 2), (0, 4), (2, 4), (4, 4)]

等价于：

>>> L = []
>>> for x in range(5):
...     if x % 2 == 0:
...         for y in range(5):
...             if y % 2 == 1:
...                 L.append((x, y))
>>> L
[(0, 1), (0, 3), (2, 1), (2, 3), (4, 1), (4, 3)]