python生成式

1.列表生成式

# 需求：生成一个列表，列表元素分别为[1**1,2**2,3**3...n**n]
li = []
for i in range(1,8):
     li.append(i**i)
print(li)

"""
# 列表生成式使用规则如下
[experssion for item in 序列 if 判断语句]
"""
# 以下两句效果等同上
li = [i**i for i in range(1,8)]
print(li)

# 可以添加判断语句，生成1~10的偶数
li = [i for i in range(1,11) if i % 2 ==0]
print(li)

#  s1='ABC' s2='123'
# A1 A2 A3 B1 B2 B3 C1 C2 C3
#可以有两个循环等同于for嵌套for
print([i + j for i in 'ABC' for j in '123'])

"""
 需求:讲3x3的矩阵转换成一堆数组(列表)
"""
li = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

resli = []
for item in li:  # [1,2,3] [4,5,6] [7,8,9]
    for item2 in item:
        resli.append(item2)
print(resli)
#生成式
resli2 = [item2 for item in li for item2 in item]
print(resli2)

 

# 1.找出1~10之间得所有偶数,并且返回一个列表(包含以这个偶数为半径得圆得面积)
# 方法一:
import math

li = []
for r in range(2, 11, 2):
    square = math.pi * r * r
    li.append(square)
print(li)

#方法二:生成式
print([math.pi * r * r for r in range(2, 11, 2)])

# 方法三:生成式中也可以调用函数
def square(r):
    """求以r为半径得圆"""
    res = math.pi * r * r
    return res


print([square(r) for r in range(2, 11, 2)])

# 找出1~100之间得所有得质数(函数+列表生成式)
"""
什么是质数:只能被1和本身整除得数,就是质数
判断num是否为质数,如果是质数,返回True 否则返回False
"""

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-BAT(T)-在线编程题)
- 题目描述:
给定一个正整数，编写程序计算有多少对质数的和等于输入的这个正整数，并输出结果。输
入值小于1000。
如，输入为10, 程序应该输出结果为2。（共有两对质数的和为10,分别为(5,5),(3,7)）
# [2,3,5,7]
- 输入描述:
输入包括一个整数n,(3 ≤ n < 1000)
- 输出描述:
输出对数
- 示例1 :
```
输入:
    10
输出:
    2
"""
#接受一个输入
num = int(input())

# 1.判断2~num之间有多少个质数
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)]
"""
1.先从列表中拿出两个数
2.判断两个数之和是否等于num
"""
#计数器
primecount = 0

#两层循环方式
# for item1 in primeli:
#     for item2 in primeli:
#         if item1 + item2 == num and item1 <= item2:
#             primecount += 1
# print(primecount)

#优化如下
# [2,3,5,7]  # 2 10-2=8 3 10-3=7 5 10-5=5 7=10-7
for item in primeli:
    if (num - item) in primeli and item <= num - item:
        primecount += 1
print(primecount)

# 也可以通过生成式将满足条件的质数对存放在另一个列表，打印列所有表元素个数
prime_pair = [[i, n - i] for i in li if i <= n - i and n - i in li]
print(len(prime_pair))

 

2.字典生成式

# 需求1:假设有20个学生,学生分数在60~100之间,筛选出成绩在90分以上得学生
import random

# 先通过随机数模拟生成20个学生信息，存放到字典中
stuInfo = {}
for i in range(20):
    name = 'westos' + str(i)
    score = random.randint(60, 100)
    stuInfo[name] = score
print(stuInfo)

# 通过生成式生成
stuInfo1 = {'westos' + str(i): random.randint(60, 100) for i in range(20)}
print(stuInfo1)

# 筛选大于90方法
highscore = {}
for name, score in stuInfo.items():
    if score > 90:
        highscore[name] = score
print(highscore)

# 通过生成式筛选出大于90的学生
print({name: score for name, score in stuInfo.items() if score > 90})

 

# 需求2:讲所有得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()})

# 需求3:大小写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)

#生成式，通过get分别取大小写K的值，没有就返回0，相加达到效果
print({k.lower(): d.get(k.lower(), 0) + d.get(k.upper(), 0) for k in d})

 

3.集合生成式

集合生成式用法于上类似，只是集合无重复性，所以不会重复

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