python-day04-列表生成式，字典生成式

一、列表生成式

1.列表生成式的入门

# 需求1:接收变量 k a b
s = '51 5000 10000'
a = s.split()
print(a）
li = []
#for循环
for item in s.split():
    li.append(int(item))
print(li)
k, a, b = li
print(k, a, b)
#列表生成式
li = [int(item) for item in s.split()]
print(li)
k,a,b = li
print(k,a,b)

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

print([i**i for i in range(1,10)]) #将0-9的幂次方生成列表打印出来
print([i**i for i in range(1,10) if i % 2 ==0])  #将0-9中的偶数幂次方生成列表打印出来

需求3：
s1 = 'ABC' S2 = '123'
要求打印出 A1 A2 A3 B1 B2 B3....

程序实现：
s1='ABC'
s2='123'
print([i+j for i in s1 for j in s2 ])

结果：

2.列表生成式的for循环

需求：
有一个嵌套列表：
[
[1,2,3],
[4,5,6],
[7,8,9]
]
将它生成[1,2,3,4,5,6,7,8,9]

#利用for循环
li=[[1,2,3],[4,5,6],[7,8,9]]
list=[]
for i in li:
    for j in i:
        list.append(j)
print(list)

#利用列表生成式
li=[[1,2,3],[4,5,6],[7,8,9]]
print([j for i in li for j in i])

#利用内置函数
from itertools import chain
li=[[1,2,3],[4,5,6],[7,8,9]]
print(list(chain(*li))) #必须得加*，否则和原来一样

结果：

牛刀小试：
练习1：
找出1～10之间的所有偶数 并返回一个列表(包含以这个偶数为半径的圆的面积)

import math #为了使用pai,则需要导入包
法一：for循环
import math
list=[]
for i in range(5):
    square=math.pi*i*i
    list.append(square)
print(list)
法二：列表生成式
import math
print([math.pi * i * i for i in range(2, 11, 2)])

结果：
练习2：
找出1-100之间的所有素数

def fun(x,y):
    list = []
    for i in range(x, y+1):
        for j in range(x+1, i):
            if i % j == 0:
                break
        else:
            list.append(i)
    return list
print(fun(1,100))

练习3：
题目需求:
对于一个十进制的正整数， 定义f(n)为其各位数字的平方和，如:
f(13) = 1**2 + 3**2 = 10
f(207) = 2**2 + 0**2 + 7**2 = 53
下面给出三个正整数k，a, b,你需要计算有多少个正整数n满足a<=n<=b,
且k*f(n)=n
输入:
第一行包含3个正整数k，a, b, k>=1, a,b<=10**18, a<=b;
输出：
输出对应的答案;

范例:
输入: 51 5000 10000
输出: 3

法一：
def f(n):
    num = str(n)
    sum = 0
    for i in num:
        sum += int(i) ** 2
    return sum
k = int(input('请输入k值：'))
while k < 1:
    print('范围有误，重新输入：')
    k = int(input('请输入k值：'))
a = int(input('请输入a值：'))
b = int(input('请输入b值：'))
while a > 10 ** 18 or b > 10 ** 18 or a > b:
    print('shuru有误，重新输入：')
    a = int(input('请输入a值：'))
    b = int(input('请输入b值：'))
cc = 0
for gg in range(a, b + 1):
    if k * f(gg) == gg:
        cc += 1
print(cc)

结果：

法二：
def f(n):
    # 1.把数字转换成字符串
    n = str(n)
    # 2.计算字符串中的每个数的平方
    sum = 0
    for item in n:
        sum += int(item) **2
    return sum
# 1.接收变量  k a b
s = input('')
li = []
for item in s.split():
    li.append(int(item))
k,a,b = li
# 2.进行判断是否满足条件
count = 0
for i in range(a,b+1):
    if k * f(i) == i:
        count += 1
print(count)

结果：

练习4：
(2018-腾讯-在线编程题)

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

num=int(input("请输入一个数："))
def prime(num):
    for i in range(2, num):
        if num % i == 0:
            return False
    else:
        return True
list=[j for j in range(1,num+1) if prime(j)]
print('%s的所有质数为：%s' %(num,list))
count=0
for item1 in list:
    if num-item1 in list and item1<=num-item1:
        count+=1
print('满足条件的数有%s个：' %(count))

结果：

二、字典生成式

需求1:假设有20个学生，学生的分数在60～100之间，筛选出成绩在90 分以上的学生
需求2：将所有的key值都变成大写
需求3：大小写的key值合并，统一以小写输出

需求1：

import random
dict = {}
for i in range(20):
    name = 'we' + str(i) #字符串的连接
    score=random.randint(60,100) #随机产生分数
    dict[name]=score  #往字典里填入值
print(dict)

法一：
highdict={}   #创建一个新的字典，用于存放90分以上的
for k,v in dict.items():
    if v>90:   #如果值大于90
        highdict[k]=v  C#将原字典的k值和value值存放新的字典
print(highdict)

法二：用字典生成式
print({k:v for k,v in dict.items() if v>90})

结果：
需求2：将所有的k值变为大写

d = dict(a = 1,b=2)
print(d)
new_d = {}
用for循化：
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值合并，统一以小写输出

for循环：
dict = dict(a=1, b=2, c=3, B=8, A=10)
dict_new={}
for k,v in dict.items():
    k_new=k.lower()
    if k_new not in dict_new:
        dict_new[k_new]=v
    else:
        dict_new[k_new]+=v
print(dict_new)
列表生成式：
d = dict(a=1, b=2, c=3, B=8, A=10)
print({k.lower(): d.get(k.upper(), 0) + d.get(k.lower(), 0) for k, v in d.items()})