Python初学者第二十四天 函数进阶(3)生成器

时间:2023-03-09 18:13:03
Python初学者第二十四天 函数进阶(3)生成器

24day

1、列表生成式:

循环模式:[变量(加工后的变量) for 变量 in iterable]

print([i for i in range(0,101,2)])
[1,4,9,16,25,36,49]
print([i*i for i in range(1,8)])
#['python1期', 'python2期', .....'python20期']
print(['python%s期'%i for i in range(1,21)])

筛选模式:[变量(加工后的变量) for 变量 in iterable if 条件]

[变量(加工后的变量) for 变量 in iterable if 条件]
print([i for i in range(1,31) if i % 3 == 0])
print([i for i in range(1,31) if i % 3 == 0])
30以内能被2整除的数的平方
print([ i*i for i in range(1,31) if i % 2 == 0])
[1,2,3,4,6,7,8]
print([i for i in range(1,9)if i != 5])
l1 = ['wusir','ba', 'aa' ,'alex']
过滤掉长度小于3的字符串列表,并将剩下的转换成大写字母
print([i.upper() for i in l1 if len(i) > 3])
['WUSIR','ALEX']
names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],
['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry', 'Eva']]
将列表中的至少含有两个'e'的人名留下来。
l1 = []
for l in names: #正常情况
for name in l:
if name.count('e') >= 2:
l1.append(name)
print(l1)
print([j for i in names for j in i if j.count('e')==2])  #列表生成式

 >>> a = [i+1 for i in range(10)]
>>> a
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> a = [i+1 for i in range(10)]
>>> a
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> a = [i+1 for i in a]
>>> a
[2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
>>> a = [i+1 for i in a]
>>> a
[3, 4, 5, 6, 7, 8, 9, 10, 11, 12]

通过列表生成式,我们可以直接创建一个列表

2、生成器:generator

生成器特性:

- 不占内存

- 只能向前,不能返回

- 当生成器达到要求会报错

创建generator,方法一:一个列表生成式的[]改成(),就创建了一个generator

>>> a3 = (i for i in range(5))
>>> a3
<generator object <genexpr> at 0x0000000001DF5BA0>
>>> next(a3)
0
>>> next(a3)
1
>>> next(a3)
2
>>> next(a3)
3
>>> next(a3)
4
>>> next(a3)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
StopIteration

通过for循环来调用生成器

>>> a3 = (i for i in range(5))
>>> for i in a3:
... print(i)
...
0
1
2
3
4

通过while循环来调生成器(跳出循环会报错)

>>> a3 = (i for i in range(5))
>>> while True:
... next(a3)
...
0
1
2
3
4
Traceback (most recent call last):
File "<stdin>", line 2, in <module>
StopIteration

方法二:使用yield创建生成器(一个函数中包含yield关键字,那么这个函数就不在是一个普通的函数,而是一个generator)

generator和函数不一样。函数是按顺序执行,遇到return语句或最后一行函数语句就返回。而变成generator的函数,在每次调用next()的时候执行,遇到yield语句返回,再次执行时会从上一次返回的yield语句处继续执行。

return VS yield

return :返回并中止function

yield :返回数据,并冰冻当前的执行过程。next唤醒冻结的函数执行过程,直到遇到下一个yield

 def fib(max):
n,a,b = 0,0,1
while n < max:
yield b #把函数的执行过程冻结在这一步,并且把b的值,返回给外面的next
print(b)
a,b = b,a+b
n += 1
return 'done' f = fib(10)
next(f)
next(f)
next(f)

当函数里面存在同时存在yield和return时,return值无法返回

 def range2(n):
count = 0
while count < n:
print('count',count)
count += 1
yield count
return 33333 new_range = range2(3)
n1 = next(new_range)
n2 = next(new_range)
n3 = next(new_range)
n4 = next(new_range)

#执行结果如下

Traceback (most recent call last):
count 0
count 1
count 2
File "D:/编辑的文本/生成器.py", line 27, in <module>
n4 = next(new_range)
StopIteration: 33333

3、生成器send

 def range2(n):
count = 0
while count < n:
print('count',count)
count += 1
sign = yield count
print('sign',sign)
return 33333 new_range = range2(3)
n1 = next(new_range)
new_range.send('stop')

#执行结果:

count 0
sign stop
count 1

send------>>>>a、唤醒并继续执行;b、发送一个信息到生成器内部;

总结:

列表推导式 与生成器表达式区别;
1,列表推导式 一行搞定,构建简单。
2, 不能用debug排错,比较复杂的列表列表推导式不能推导出来。 生成器表达式:
1,循环模式。 (变量(加工后的变量) for 变量 in iterable)
(变量(加工后的变量) for 变量 in iterable if 条件)
1,一行搞定。
2,节省内存。