迭代器 & 生成器

迭代器

访问集合元素的一种方式，迭代器对象从集合的第一个元素开始访问，直到素有的元素被访问完结束，迭代器只能往前不会后退。不要求事先准备好整个迭代过程中所有的元素，迭代器仅仅在迭代到某个元素时才计算该元素，而在这之前或之后，元素可以不存在或者被销毁。适用于遍历一些巨大的或是无限的集合，比如几个G的文件。

迭代器特点

访问者不需要关心迭代器内部的结构，仅需通过 next() 方法不断去取下一个内容
不能随机访问集合中的某个值，只能从头到尾一次访问
访问到一半时不能往回退
便于循环比较大的数据集合，节省内存

生成一个迭代器

nums = iter([1,2,3])
print(nums)
print(nums.__next__()) # 1， Python 2.7 nums.next()
print(nums.__next__()) # 2
print(nums.__next__()) # 3
print(nums.__next__()) # StopIteration

生成器

一个函数调用时返回一个迭代器，那个函数就叫做生成器(generator)，如果函数中包含yield 语法，那这个函数就会变成生成器

只要含有 yield 关键字的函数都是生成器函数
yield 和 return 不能共用且需要卸载函数内
只要是生成器函数执行之后得到一个生成器作为返回值
生成器.__next__() 返回生成器执行结果，生成器也是迭代器

yield 作用

yield 可以是函数中断，并保存中断状态，中断后，代码可以继续往下执行，过一段时间还可以在重新调用这个函数，从上次 yield 的下一句开始执行。

def cash_out(amount): # 函数是生成器
  while amount > 0:
    amount -= 100
    yield 100
    print("擦，又来取钱了...败家子")

ATM = cash_out(500) # ATM 是迭代器
print(type(ATM)) # <class 'generator'>
print("取到钱 %s 万" ATM.__next__()) # 100万
print("取到钱 %s 万" ATM.__next__()) # 100万
print("取到钱 %s 万" ATM.__next__()) # 100万
print("取到钱 %s 万" ATM.__next__()) # 100万
print("取到钱 %s 万" ATM.__next__()) # 100万

总结

双下方法：很少直接调用，一般通过其他语法触发此方法
可迭代的 - 可迭代协议：含有iter的方法('iter' in dir(数据))
可迭代的一定可以被 for 循环
迭代器协议：含有iter和next方法
迭代器一定可迭代，可迭代器的通过调用iter()方法就能得到一个迭代器
迭代器特点
- 方便使用，切只能取所有的数据取一次
- 节省内存空间
生成器
- 生成器的本质就是迭代器
- 生成器表现形式
  - 生成器函数
  - 生成器表达式
生成器函数
- 含有 yield 关键字的函数就是生成器函数
- 调用函数之后函数不执行，返回一个生成器
- 函数外调用next方法时候渠道一个值
- 直到取完最后一个，在执行next会报错

def generator():
  for i in range(2000000):
    yield '哈哈哈%s' %i
g = generator() #调用生成器函数得到一个生成器
res = g.__next__() # 每一次执行g.__next__就是从生成器中取值，预示着生成器函数中的代码继续执行
print(res)

num = 0
for i in g:
  num += 1
  if num > 50:
    break
  print(i)

从生成器中取值的几个方法
- next 方法
- for
- 数据类型的强制转换（占用所有数据的内存）
send 获取下一个值得效果和 next 基本一样，只是在获取下一个值的时候，给上一个值得位置传递一个数据
- 第一次使用生成器的时候，用next 获取下一个值
- 最后一个yield 不能接受外部的值

生成器表达式

每次只能获取一个数据，不占用太多内存

g = (i for i in range(10))
print(g)
for i in g:
  print(i)

列表推导式

获取所有数据，占用内存

egg_list = ['鸡蛋%s' %i for i in range(10)]
print(egg_list)

各种推导式

[每一个元素或者是和元素相关的操作 for 元素 in 可迭代数据类型] #遍历之后挨个处理

[满足条件的元素相关的操作 for 元素 in 可迭代数据类型 if 元素相关的条件] #筛选功能

#30以内所有能被3整除的数
ret = [i for i in range(30) if i%3 == 0]  #完整的列表推导式
g = (i for i in range(30) if i%3 == 0)  #完整的列表推导式
print(ret)

#30以内所有能被3整除的数的平方
ret = [i*i for i in (1,2,3,4) if i%3 == 0]
ret = (i*i for i in range(30) if i%3 == 0)
print(ret)

# 例三:找到嵌套列表中名字含有两个‘e’的所有名字
names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],
#['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry', 'Eva']]
ret = [name for lst in names for name in lst if name.count('e') ==2]
ret = (name for lst in names for name in lst if name.count('e') ==2)
print(ret)

字典推导式

# 例一：将一个字典的key和value对调
mcase = {'a': 10, 'b': 34}
#{10:'a' , 34:'b'}
mcase_frequency = {mcase[k]: k for k in mcase}
print(mcase_frequency)

# 例二：合并大小写对应的value值，将k统一成小写
mcase = {'a': 10, 'b': 34, 'A': 7, 'Z': 3}
#{'a':10+7,'b':34,'z':3}
mcase_frequency = {k.lower(): mcase.get(k.lower(), 0) + mcase.get(k.upper(), 0) for k in mcase}
print(mcase_frequency)

集合推导式，自带结果去重功能

squared = {x**2 for x in [1, -1, 2]}
print(squared)

各种推导式：生成器列表字典集合
遍历操作
筛选操作

Previous面向对象编程 Next装饰器

Last updated 5 years ago

Was this helpful?