Python 列表生成式和字典生成式

列表生成式

如果要生成[1x1, 2x2, 3x3, …, 10x10]怎么做？方法一是循环：

>>> L = []
>>> for x in range(1, 11):
...    L.append(x * x)
...
>>> L
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

如果我们想要生成一个列表，而使用循环的话又太繁琐，刚好Python有一个强大的功能可以用一行语句代替循环。

>>> [x * x for x in range(1, 11)] 
[1, 4, 9, 16, 25, 36, 49, 64, 81, 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()]
['y=B', 'x=A', 'z=C']

最后把一个list中所有的字符串变成小写：

>>> L = ['Hello', 'World', 'IBM', 'Apple']
>>> [s.lower() for s in L]
['hello', 'world', 'ibm', 'apple']

if … else

使用列表生成式的时候，我们不能在最后的if加上else,这是因为跟在for后面的if是一个筛选条件，不能带else，否则如何筛选？

在一个列表生成式中，for前面的if … else是表达式，而for后面的if是过滤条件，不能带else。

字典生成式

字典生成式和列表类似。

zodiac = '猴鸡猪狗鼠牛虎兔龙蛇马羊'

year = 2020

#计算年龄
birth_date = int(input("请输入出生日期："))
age = year - birth_date
print(age)

#计算生肖
aaa = birth_date % 12
shuxiang = zodiac[aaa] #属相
print(shuxiang)

#存入字典
shuxiang_dict = {a:age for a in shuxiang}

print(shuxiang_dict)