Python之生成式

一、列表生成式

示例1：
1、非列表生成式

s = '51 5000 10000'
li = []
for item in s.split():
    li.append(item)

k,a,b = li
print(k,a,b)

2、使用列表生成式

s = '51 5000 10000'
li = []
k,a,b = [int(item) for item in s.split()]
print(k,a,b)

示例2:生成一个列表，列表元素分别[12,22,32…n2]
1、不使用列表生成式

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

2、使用列表生成式

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

二、列表生成式小练习

1、

需求：

找出1～10之间所有偶数，返回一个列表，包含以这个偶数为半径
的圆的面积

代码：

import math

a = math.pi
print([i * i * a for i in range(2,11,2)])

2、

需求：

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

代码示例：

def isPrime(num):
    for i in range(2,num):
        if num % i == 0:
            return False				#列表生成式只能判断一个条件是True还是False
    else:
        return True

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

3、

找出/var/log目录下以.log结尾的文件

代码示例：

import os

print([filename for filename in os.listdir('/var/log') if filename.endswith('.log')])

4、

需求：

给定一个正整数，编写程序计算有多少对质数的和等于输入的这个正整
数，并输出结果。输
入值小于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)]
print(primeli)
#1.先从列表里拿出两个数
#2.判断两个数之和是否等于num
primecount = 0
for item in primeli:
    if (num - item) in primeli and item <= num - item:
        primecount += 1
print(primecount)

三、字典生成式

一、

需求：

假设有20个学生，学生名为redhatX，学生成绩在60～100之间，
筛选出成绩在90分以上的学生

1、不使用字典生成式

代码示例：

import random

StuInfo = {}
Higscore = {}

for i in range(20):
    name = 'redhat' + str(i)
    score = random.randint(60,100)
    StuInfo[name] = score

print(StuInfo)

for i,v in StuInfo.items():
    if v > 90:
        Higscore[i] = v

print(Higscore)

2、使用字典生成式

代码示例：

import random

StuInfo = {}
Higscore = {}

for i in range(20):
    name = 'redhat' + str(i)
    score = random.randint(60,100)
    StuInfo[name] = score

print(StuInfo)

print({i:v for i,v in StuInfo.items() if v > 90})

二、

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

代码示例：

d = dict(a=1,b=2)
print({k.upper():v for k,v in d.items()})

三、将字典里的大小写、value值合并，统一以小写输出

代码示例：

d = dict(a=2,b=1,c=2,B=9,A=10)

print({k.lower():d.get(k.lower(),0) + d.get(k.upper(),0) for k in d})