python的列表生成式和字典生成式

列表生成式

1. 什么是列表生成式

列表生成式即List Comprehensions，是Python内置的非常简单却强大的可以用来创建list的生成式
列表生成式也可以叫做列表解析
列表生成式的格式：[expression for i in 序列 if…]==表达式+循环+条件

一般情况下循环太繁琐，而列表生成式则可以用一行语句代替多行循环生成列表
运用列表生成式，可以写出非常简洁的代码
运用列表生成式，可以快速生成list，可以通过一个list推导出另一个list，而代码却十分简洁
1
2
3

2. 用列表的append方法生成列表

使用for循环的方法：

s = '51 5000 10000'
 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
3
4
5
6
7

生成一个列表，列表元素分别为[12,22,32…n2]

li = []
# for i in range(1,10):
#     li.append(i ** 2) #打印1～9的平方
# print(li)
# print([i ** 2 for i in range(1,10)])
1
2
3
4
5
6

2. 用列表生成式生成一个列表

[expression for i in 序列 if...] #带条件的列表生成式
列表元素为1～10中偶数的平方

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

3. 列表生成式练习1：用列表生成式生成一个列表，包含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)])
1
2
3
4
5
6
7

4. 列表生成式练习2：用列表生成式生成一个列表，包含1到num的所有质数

给定一个正整数，编写程序计算有多少对质数的和等于输入的这个正整
数，并输出结果。输
入值小于1000。
如，输入为10, 程序应该输出结果为2。（共有两对质数的和为10,分别
为(5,5),(3,7)）
[2,3,5,7]

• 输入描述:
  输入包括一个整数n,(3 ≤ n < 1000)

• 输出描述:
  输出对数

• 示例1 :

输入:
    10
输出:
    2
"""
1
2
3
4
5

num = int(input())  #传入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)] #拿到2到num的所有质数
print(primeli)
"""
1.先拿出两个数
2.判断两个数之和是否等于num
"""
count = 0  #定义两质数之和为num的计数器为0
for item1 in primeli:
    if (num - item1) in primeli and item1 <= num - item1:  #这里要去掉重复的
        count += 1  #发现符合要求,计数器加1
print(count)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

字典生成式

选出20个学生中成绩在90分以上的学生

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

import random  #定义一个生成随机数的包
 stuInfo = {}  #定义一个空字典，用来存储学生信息
 for i in range(20):  #i从0到19
     name = 'westos' + str(i) 学生名称为westos加数字，将数字类型转为字符串类型才能与westos连接
     score = random.randint(60,100)  #随机生成60到100的整数作为学生的分数
     stuInfo[name] = score   #给每个name赋value值
 # print(stuInfo)  
  highscore = {}  #定义一个字典，存储高分（成绩在90分以上）
  for name,score in stuInfo.items():  #查看stuInfo字典的name和score键值对
     if score > 90:  #如果分数大于90
         highscore[name] = score  #就将分数赋给highscore字典
 print(highscore)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

字典生成式

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

示例2：将列表key值变为大写

d = dict(a=1,b=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)
1
2
3
4
5
6
7
8

可以看到，key值的小写字母变成了大写字母
字典生成式

print({k.upper(): v for k,v in d.items()})
1

示例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() #将键的大写变成小写赋给low_k变量
     if low_k not in new_d:  如果low_k的元素不在new_d
         new_d[low_k] = v  将对应的value值给new_d
     else:
         new_d[low_k] += v  #否则就给原有的元素加上value值
 print(new_d)
1
2
3
4
5
6
7
8
9
10
11
12

字典生成式

print({k.lower(): d.get(k.lower(),0) + d.get(k.upper(),0) for k in d})   #打印小写字母：通过键来获得大小写字母的value值，并且返回i0
1

练习

示例

li = [     #定义一个列表，这个列表的每一个元素都是一个列表
    [1,2,3],
    [4,5,6],
    [7,8,9]
]
#[1,2,3,4,5,6,7,8,9]
result = []
# for i in li:    #[1,2,3],[4,5,6],[7,8,9]
#     for j in i:
#         result.append(j)
# print(result)
1
2
3
4
5
6
7
8
9
10
11
12
13

print([j for i in li for j in i])
from itertools import chain
print(list(chain(*li)))

• 1
• 2
• 3
• 4
• 5
• 6