小菜鸟的python进阶之路 ------- 生成式

生成式

• 集合生成式

• 列表生成式

• 字典生成式

 

1.集合生成式

• 格式：{ expression for i in 序列 if 条件 }
• 作用：生成一个集合

示例：

print({i ** 2 for i in {1,2,3,3}})
print({i ** 2 for i in {1,2,3,9,12,9} if i % 3 == 0 })

2.列表生成式

• 格式：[expression for i in 序列 if 条件]
• 作用：生成一个列表

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

print([i**2 for i in range(n)])

练习1.找出1-10之间的所有偶数，并返回一个列表(包含以这个偶数为半径的圆的面积)

import math
a = math.pi

第一种方式输出：
print([i**2*a for i in range(1,11) if i % 2 == 0])

第二种方式输出:
# print([math.pi * r * r for r in range(2,11,2)])

练习2.找出1～100之间所有的质数

def isPrime(num):
    for i in range(2,num):
        if num % i == 0:
            return False
    else:
        return True

print([i for i in range(2,101) if isPrime(i)])

3.字典生成式

• 格式：{expression for i in 序列 if 条件}
• 作用：生成一个字典

示例：输入:d=dict(a=1,b=2)  ，输出:d={'A'=1,'B'=2}

#第一种方式：
d = dict(a=1,b=2)
print(d)

new_d = {}
for i in d:
    new_d[i.upper()] = d[i]

print('key转化为大写的字典:',new_d)



#字典生成式生成的方式
print({k.upper():v for k,v in d.items()})

练习：d = dict(a=2,b=1,c=2,B=9,A=10)，输出{'a':12,'b':10,'c':2}
 

第一种方法：
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)


第二种方法：
d = dict(a=2,b=1,c=2,B=9,A=10)
print({ k.lower():(d.get(k.lower(),0)+d.get(k.upper(),0)) for k in d})