python推导式
推导式是用一行式子来完成循环操作的语句，一般与for循环结合来使用。
列表推导式
公式
[ expr for value in collection [if condition] ]
例子
对循环内元素的选择
x = [i for i in range(100) if (i % 2) != 0 and (i % 3) == 0]
print(x)
# [3, 9, 15, 21, 27, 33, 39, 45, 51, 57, 63, 69, 75, 81, 87, 93, 99]
对循环(列表)内元素的操作
x = [i ** 2 for i in range(1, 10)]
print(x)
# [1, 4, 9, 16, 25, 36, 49, 64, 81]
y = [-4, -2, 0, 2, 4]
z = [a * 2 for a in y]
print(z)
# [-8, -4, 0, 4, 8]
列表嵌套元组(一个循环)
x = [(i, i ** 2) for i in range(6)]
print(x)
# [(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25)]
列表嵌套元组(两个循环)
a = [(i, j) for i in range(0, 3) for j in range(0, 3)]
print(a)
# [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]
两个循环再加选择
a = [(i, j) for i in range(0, 3) if i < 1 for j in range(0, 3) if j > 1]
print(a)
# [(0, 2)]
列表嵌套列表与上面类似
x = [[i, j] for i in range(0, 3) for j in range(0, 3)]
print(x)
# [[0, 0], [0, 1], [0, 2], [1, 0], [1, 1], [1, 2], [2, 0], [2, 1], [2, 2]]
元组推导式
公式与列表一样
( expr for value in collection [if condition] )
例子
a = (x for x in range(10))
print(a)
# <generator object <genexpr> at 0x0000025BE511CC48>
print(tuple(a))
# (0, 1, 2, 3, 4, 5, 6, 7, 8, 9)
与列表不同的是，只是输出a我们得到的只是一个生成器，还需要对其进行转换，这里运用到tuple()函数，它可以将生成器对象转换为元组。
而在对于生成器对象，如果我们对它进行输出,则元组中的元素也会消失如下：
a = (i for i in range(10)) # 生成一个元组
for i in a: # for 循环遍历生成器对象
print(i,end=',') # 输出元组元素在同一行显示，并且用“,”隔开
print(tuple(a)) # 输出新元组
#0,1,2,3,4,5,6,7,8,9,()
还可以通过next()函数来对元组推导式进行输出
e = (i for i in range(10))
print(e)
# <generator object <genexpr> at 0x0000007A0B8D01B0>
print(next(e)) # 0
print(next(e)) # 1
for each in e:
print(each, end=' ')
# 2 3 4 5 6 7 8 9
同样输出后元组推导式中的元素就消失了。
字典推导式
公式
{ key_expr: value_expr for value in collection [if condition] }
例子
b = {i: i % 2 == 0 for i in range(10) if i % 3 == 0}
print(b)
# {0: True, 3: False, 6: True, 9: False}
集合推导式
公式
{ expr for value in collection [if condition] }
例子
c = {i for i in [1, 2, 3, 4, 5, 5, 6, 4, 3, 2, 1]}
print(c)
# {1, 2, 3, 4, 5, 6}
推导式求和
如下，运用sum()函数，即可实现。
s = sum([i for i in range(101)])
print(s) # 5050
s = sum((i for i in range(101)))
print(s) # 5050