各种推导式,主要使用示例演示用法
列表生成式
示例1:求0-9每个数的平方
li = [x*x for x in range(10)]
print(li)
# 执行结果:
# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
列表生成式,使用[],括号内是for循环语句
示例2:求0-9能被3整除的数字
li = [x for x in range(10) if x % 3 == 0]
print(li)
# 执行结果:
# [0, 3, 6, 9]
在这个列表生成式中,不断有for循环,而且还用到了if判断语句
列表生成式的格式:
(1)[元素 for 元素 in 可迭代的类型 ]
(2)[元素 for 元素 in 可迭代的类型 if 元素需要满足的条件]
生成器表达式
还是上面两个例子,使用生成器表达式写出来:
示例1:求0-9每个数的平方
g = (x for x in range(10) if x % 3 == 0)
print(g) # generator 这里的g就是一个生成器了,将中括号换成小括号就变成了生成器表达式
for i in g:
print(i)
# 执行结果:
# at 0x00000194C9FA34C0>
# 0
# 3
# 6
# 9
示例2:求0-9能被3整除的数字
g = (x for x in range(10) if x % 3 == 0)
print(next(g))
print(next(g))
print(next(g))
print(next(g))
# 执行结果:
# 0
# 3
# 6
# 9
列表生成式和生成器表达式总结:
1. 把列表生成式的中括号换成小括号就是生成器表达式
2. 列表解析与生成器表达式都是一种便利的编程方式,只不过生成器表达式更节省内存
示例3:找到嵌套列表中名字含有两个'e'的所有名字
names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],
['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry', 'Eva']]
# for name in names:
# for i in name:
# if i.count('e') == 2:
# print(i)
# 如果无法一次写出嵌套的列表生成式,可以先使用for循环写出来
list_name = [x for name in names for x in name if x.count('e') == 2]
print(list_name)
# 执行结果:
# ['Jefferson', 'Wesley', 'Steven', 'Jennifer']
字典推导式
示例1:将字典中的key和value对调
# 注意字典推导式中,返回的一定是一个字典类型:mcase[i]:i
dic = {mcase[i]:i for i in mcase}
print(dic)
# 执行结果:
# {10: 'a', 34: 'b'}
示例2:合并大小写对应的value值,将k统一成小写
mcase = {'a': 10, 'b': 34, 'A': 7, 'Z': 3}
dic = {k.lower(): mcase.get(k.lower(), 0) + mcase.get(k.upper(), 0) for k in mcase}
print(dic)
# 执行结果:
# {'a': 17, 'b': 34, 'z': 3}
# 这个示例中用到了 get() 这个知识点:
# mcase.get(k.lower(), 0) 当字典mcase中存在k.lower()的时候则直接返回,否则返回默认填写的0
集合推导式
集合数据类型,有自带的去重功能
squared = {x**2 for x in [1, -1, 2]} # 返回 1, 1, 2 集合自带去重的功能
print(squared)
# 执行结果:
# {1, 4}
其他一些示例:
# 例1: 过滤掉长度小于3的字符串列表,并将剩下的转换成大写字母
li = ['a', 'abc', 'c', 'dddd']
l1 = [i.upper() for i in li if len(i) >= 3]
print(l1)
# 例2: 求(x,y)其中x是0-5之间的偶数,y是0-5之间的奇数组成的元祖列表
l1 = [(x, y) for x in range(1, 5) if x%2==0 for y in range(1, 5) if y%3==0]
print(l1)
# 例3: 求M中3,6,9组成的列表M = [[1,2,3],[4,5,6],[7,8,9]]
M = [[1,2,3],[4,5,6],[7,8,9]]
l1 = [i[2] for i in M]
print(l1)