列表推导是一种从其他列表创建列表的方式,类似于数学中的集合推导
注意:在列表推导中,for前面只有一个表达式
# 生成[1x1, 2x2, 3x3, ..., 10x10]
[x * x for x in range(1, 11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
# for循环后面还可以加上if判断,可以筛选出仅偶数的平方:
[x * x for x in range(1, 11) if x % 2 == 0]
[4, 16, 36, 64, 100]
# 可以使用两层循环,生成全排列:
[m + n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']
# 列表生成式可以使用两个变量来生成list:
d = {'x': 'A', 'y': 'B', 'z': 'C' }
[k + '=' + v for k, v in d.items()]
['x=A', 'y=B', 'z=C']
# 把一个list中所有的字符串变成小写:
L = ['Hello', 'World', 'IBM', 'Apple']
[s.lower() for s in L]
['hello', 'world', 'ibm', 'apple']
# 不能在最后的if加上else:
res = [i for i in range(1,11) if i % 2 == 0 else 0]
print(res)
SyntaxError: invalid syntax
# if写在for前面必须加else,否则报错:
res = [i if i % 2 == 0 for i in range(1,11)]
print(res)
SyntaxError: invalid syntax
res = [i if i % 2 == 0 else -i for i in range(1,11)]
print(res)
[-1, 2, -3, 4, -5, 6, -7, 8, -9, 10]
# 在一个列表生成式中,for前面的if ... else是表达式,而for后面的if是过滤条件,不能带else
扩展案例
# 将名字的首字母相同的男孩和女孩配对
girls = ['alice','bernice','clarice']
boys = ['chris','arnold','bob']
res = [b + '+' + g for b in boys for g in girls if b[0] == g[0]]
# ['chris+clarice', 'arnold+alice', 'bob+bernice']
print(res)