生成式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)))