Python -- 函数之推导式
5.12 推导式
l = []
for i in range(1,11):
l.append(i)
print(l)
# 用列表推导式 (一行搞定)
l = [i for i in range(1,11)]
print(l)
- 列表推导式可分为两种模式:
- 循环模式: [变量(加工的变量) for 变量 in iterable(可迭代对象)]
- 筛选模式: [变量(加工的变量) for 变量 in iterable(可迭代对象) if 条件]
# 循环模式
1. 将10以内所有整数的平方写入列表。
l = [i**2 for i in range(1,11)]
print(l)
2. 100以内所有的偶数写入列表.
l = [i for i in range(2,101,2)]
3. 从python1期到python100期写入列表lst
l = [f'python{i}期' for i in range(1,101)]
print(l)
# 筛选模式
1. 将这个列表中大于3的元素留下来。
l1 = [4, 3, 2, 6, 5, 5, 7, 8]
print([i for i in l1 if i > 3])
2. 三十以内可以被三整除的数。
l = [i for i in range(30) if i % 3 == 0]
print(l)
3. 过滤掉长度小于3的字符串列表,并将剩下的转换成大写字母
l = ['wusir', 'laonanhai', 'aa', 'b', 'taibai']
print([i.upper() for i in l if len(i) > 3])
4. 找到嵌套列表中名字含有两个‘e’的所有名字(有难度)
names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry', 'Eva']]
print([n for i in names for n in i if n.count('e') >= 2])
# 注意遍历顺序是实现的关键
-
生成器表达式
生成器表达式和列表推导式的语法上一模一样,只是把 [] 换成 () 就行了。
1. 比如将十以内所有数的平方放到一个生成器表达式中
gen = (i**2 for i in range(10))
print(gen)
# 结果是内存地址 <generator object <genexpr> at 0x000002BFC2431C50>
# 取值
for n in gen:
print(n)
2. 生成器表达式也可进行筛选
# 100 以内能被3整除的数
gen = (i for i in range(100) if i % 3 == 0)
for num in gen:
print(num)
- 生成器表达式和列表推导式的区别:
- 离别推导式比较耗内存,所有的数据一次性加载到内存,而生成器表达式遵循迭代器协议,逐个产生元素
- 得到的值不一样,列表推导式得到的是一个列表,生成器表达式获取的是一个生成器(内存地址)
- 列表推导式一目了然,生成器表达式只是一个内存地址
- 其他相关的推导式
# 字典推导式
lst1 = ['jay','jj','meet']
lst2 = ['周杰伦','林俊杰','陈奕迅']
dic = {lst1[i]:lst2[i] for i in range(len(lst1))}
print(dic)
# 集合推导式
集合推导式可以帮我们直接生成一个集合,集合的特点;无序,不重复 所以集合推导式自带去重功能
lst = [1,2,3,-1,-3,-7,9]
s = {abs(i) for i in lst}
print(s)