[Python] 列表推导式

列表推导是一种从其他列表创建列表的方式，类似于数学中的集合推导

注意:在列表推导中，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)