python-生成式

生成式Comprehensions

一般情况下循环太繁琐，而生成式可以用一行生成式代替多行循环语句

• 列表生成式List Comprehensions

列表生成式格式：[expression for i in 序列 if…] --> 表达式+循环+条件
列表生成式也叫做列表解析

 s = input()
 li = []
 for item in s.split():
     li.append(int(item))
 k,a,b = li
 
 生成式：
 k,a,b = [int(item) for item in s.split()]
 print(k,a,b)

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

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

只生成偶数的平方

print([i ** 2 for i in range(1,10) if i % 2 == 0])

生成两个字符串的轮询组成

print ([ m + n for m in 'ABC' for n in 'XYZ' ])

1~num之间所有的质数

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)])

列表生成式练习

(2018-腾讯-在线编程题)
题目描述:
给定一个正整数，编写程序计算有多少对质数的和等于输入的这个正整
数，并输出结果。输入值小于1000。
如，输入为10, 程序应该输出结果为2。（共有两对质数的和为10,分别
为(5,5),(3,7)）
[2,3,5,7]
输入描述:
输入包括一个整数n,(3 ≤ n < 1000)
输出描述:
输出对数
示例1 :
输入:
10
输出:
2

生成质数列表：

num = int(input())

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

primeli = [i for i in range(2,num) if isPrime(i)]
print(primeli)

1.质数列表里循环，num减循环到的数，差也在质数列表里
2.防止循环时，两个质数颠倒会再统计，第一个质数小于等于第二个质数

count = 0

for item1 in primeli:
    if (num - item1) in primeli and item1 <= num - item1:	两个质数都在集合里，且，第一个质数小于等于第二个质数
        count += 1

print(count)

• 字典生成式

需求：假设有20个学生，学生名为westisX，学生分数在60-100之间，选出成绩在90分以上的学生

import random    
stuInfo = {}    
for i in range(20):				生成学生和成绩
    name = 'westos' + str(i)
    score = random.randint(60,100)
    stuInfo[name] = score
print(stuInfo)

highscore = {}
for name,score in stuInfo.items():
    if score > 90:				判断成绩高于90
        highscore[name] = score
print(highscore)

print({name: score for name,score in stuInfo.items() if score > 90})

需求：将所有的key值变为大写

d = dict(a=1,b=2)
print(d)

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()})

需求：大小写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)

print({k.lower(): d.get(k.lower(),0) + d.get(k.upper(),0) for k in d}) 		0:设置不存在的key的value值为0

• 生成式变式

例：

li = [
[1,2,3],
[4,5,6],
[7,8,9]
]

输出：[1,2,3,4,5,6,7,8,9]

result = []

for i in li:    
    for j in i:
        result.append(j)
print(result)

print([j for i in li for j in i])

from itertools import chain

print(list(chain(*li)))