迭代器 、生成器

# 迭代器 、生成器

## 迭代器

### 迭代器定义

- 迭代器是一个对象,帮助你从某种对象(可迭代对象 如:str/list/tuple/dict/set)中的元素逐一获取,平时我们用的for循环就是利用的迭代器

### 怎么判断是不是迭代器

- 具有__next__()函数就是一个迭代器

### 把可迭代对象转换成迭代器并输出

- 利用内置函数iter()创建迭代器v=iter(v1),或者v1.__iter__(),迭代器可以利用__next__()方法获取元素

  #利用iter()函数生成迭代器v,用__next_()函数获取迭代器里面的元素
  v1= [11,22,33]
  v = iter(v1)   #生成迭代器v
  print(v)  
  while True:
      try:
          result = v.__next__()  #获取迭代器的值
          print(result)
      except:
          pass
  结果:
  <list_iterator object at 0x00000000021CF248>
  11
  22
  33
  

### 可迭代对象

- 可以被for循环的
- 具有__iter__()方法的

## 生成器

### 生成器定义

- 生成器是边循环边计算机制,函数中有yield,那么改函数就是一个生成器函数,调用生成器函数会返回一个生成器

  #生成器函数,调用函数时候不执行,只有用for循环才执行函数,
  # 每次遇到yield就停止,并返回yield值,下次接着上次的函数继续执行
  def func():
      print("yield前打印")
      yield 1     #第一次循环到这里返回1,并停止往下执行
      yield 2      #第二次循环接着yield 的后面执行
      yield 3
  
  v = func() #生成一个生成器
  print(v)
  for item in v:
      print(item)
      break
  结果:
  <generator object func at 0x0000000002726D48>
  yield前打印
  1
  

### 生成器的特点

#生成器函数,调用函数时候不执行,只有用for循环才执行函数,
# 每次遇到yield就停止,并返回yield值,下次接着上次的函数继续执行
def func():
    print("yield前打印")
    yield 1     #第一次循环到这里返回1,并停止往下执行
    yield 2      #第二次循环接着yield 的后面执行
    yield 3

v = func() #生成一个生成器
print(v)
for item in v:
    print(item)
    break
结果:
<generator object func at 0x0000000002726D48>
yield前打印
1

例子2:

#利用生成器输出数字
def func():
    count =0
    while True:
# 当第一次循环到这里时候停止往下执行并返回count值0
        yield count 
#当第二次循环继续count+=1,后再次回到while true ,然后第二次到达yield count,这时打印1
        count+=1
        if count==50:
            return  #当count==50时退出函数

v1 = func()  #生成生成器
for item in v1:
    print(item)
结果:
0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49


- 边生成应用,一次只生成一个元素,占用内存小
- 生成器有迭代器功能
- 函数内有yield
- 生成器函数调用时候不执行,调用时候返回一个生成器
- 生成器用for循环才执行函数
- 函数遇到yield就停止,并返回yield的值
- 二次循环接着上次停止的yield位置后面执行

### yield from应用

def func1():
    yield 88
    yield 99
def func():
    yield 1
    yield 2
    yield from func1()  #调到func1执行获取yield的值
    yield 4

v1 = func()
for item in v1:
    print(item)
结果:
1
2
88
99
4


### 生成器推导式

v2 = (i for i in range(10))  #创建生成器,注意这里不是元组
print(v2)
for i in v2:  #循环才执行
    print(i)
结果:
0
1
2
3
4
5
6
7
8
9



*XMind - Trial Version*

posted @ 2021-04-13 22:39  傻老头  阅读(71)  评论(0)    收藏  举报