python——生成式详解

集和生成式

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 })   ##** == ^
结果：
{1, 4, 9} ##集合的去重功能
{81, 9, 144}

列表生成式

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

题目：生成一个列表，列表元素分别为[12,22,32,…n2]

正常方法：

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


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

列表生成式

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

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

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

import math
a = math.pi
print(a)   ##π可以使用math.pi导入

print([math.pi * r * r for r in range(2,11,2)])
结果：
3.141592653589793
[12.566370614359172, 50.26548245743669, 113.09733552923255, 201.06192982974676, 314.1592653589793]

找出1～100之间所有的质数

def isPrime(num):
    for i in range(2,num):
        if num % i == 0:
            return False          ##只有一个能被整数的数就返回False
    else:
        return True

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

字典生成式

**题目1：**将所有的key值变为大写

d = dict(a=1,b=2)
new_d = {}
for i in d:
    new_d[i.upper()] = d[i]

print({k.upper():v for k,v in d.items()})

结果：
{'A':1,'B':2}
{'A':1,'B':2}

题目2：
将大小写key值并值，统一以小写输出
例子：
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)
结果：
{'a':12,'b':10,'c':2}

字典做法：

print({k.lower(): d.get(k.lower(),0) + d.get(k.upper(),0) for k in d})
##.get(key,x) 的用法：如果key值存在，则返回对应的value值，否则返回x

结果：
{'a':12,'b':10,'c':2}