Python中生成式的使用

列表生成式

Lst = [expression for i in ?? if...]
s = '1 2 3'
li = [int(i) for i in s.split()]
print(li)

找质数

    # 找质数
def findnumber(n):
    for i in range(2, n):
        if n % i == 0:
            return 0
    else:
        return 1


number = int(input())
li = [i for i in range(2, number+1) if findnumber(i) == 1]
print(li)

列表生成式变形

a = '123'                                  
b = 'ABC'                                  
li = [ i + j for i in a for j in b ]       
print(li)        

字典生成式

{key: value for key, value in dic.items() if value...}                 
d = dict(a=1,b=2,c=2,A=2,B=2)
print({i:d.get(i,0)+d.get(i.upper(),0) for i in d.keys() if i>='a' and i<='z' })