Python的列表生成式、字典生成式

一、列表生成式

格式
[expression for i in 序列 if …]

实例1：

#将字符串转换为整型
s = '51 5000 10000'
li = []
for item in s.split():
    li.append(int(item))
k,a,b = li
print(k,a,b)

k,a,b = [int(item) for item in s.split()]
print(k,a,b)

运行结果：

51 5000 10000
51 5000 10000

实例2：

#生成一个列表，列表元素分别为[1**2,2**2,3**2...n**2]

li = []
for i in range(1,8):
    li.append(i ** 2)
print(li)

print([i ** 2 for i in range(1,8)])

运行结果：
[1, 4, 9, 16, 25, 36, 49]
[1, 4, 9, 16, 25, 36, 49]

二、字典生成式

格式

{expression for i in 序列 if …}

实例：

#需求2：将所有key值变为大写
d = dict(a=1,b=2)
new_d = {}
for i in d:
    new_d[i.upper()] = d[i]
print(new_d)
print({k.upper():v for k,v in d.items()})

运行结果：

{‘A’: 1, ‘B’: 2}
{‘A’: 1, ‘B’: 2}

实例2：

#需求3:大小写、value值和并，统一以小写输出
d = dict(a=2,b=1,c=2,B=9,A=10)
new_d = {}
for k,v in d.items():
    low_k = k.lower()
    if low_k not in new_d:
        new_d[low_k] = v
    else:
        new_d[low_k] += v

print(new_d)

print({k.lower():d.get(k.lower(),0) + d.get(k.upper(),0) for k in d})

运行结果：

{‘a’: 12, ‘b’: 10, ‘c’: 2}
{‘a’: 12, ‘b’: 10, ‘c’: 2}

三、集合生成式