学了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)]

 

posted on 2019-03-14 16:51  程序员一学徒  阅读(1180)  评论(0编辑  收藏  举报