欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python3之手动创建迭代器的实例代码

程序员文章站 2022-06-21 09:15:25
迭代器即可以遍历诸如列表,字典及字符串等序列对象甚至自定义对象的对象,其本质就是记录迭代对象中每个元素的位置。迭代过程从第一个元素至最后一个元素,且过程不能回滚或反方向迭代...

迭代器即可以遍历诸如列表,字典及字符串等序列对象甚至自定义对象的对象,其本质就是记录迭代对象中每个元素的位置。迭代过程从第一个元素至最后一个元素,且过程不能回滚或反方向迭代。

两个基本方法iter、next

序列对象可以利用 iter() 直接创建迭代器,并通过 next() 即可迭代迭代器。

利用for循环迭代

s = 'python'
it = iter(s)
for it in it:
  print(it)

示例结果:

p
y
t
h
o
n

利用next()迭代

s = 'python'
it = iter(s)
print(next(it))
print(next(it))
print(next(it))
print(next(it))
print(next(it))
print(next(it))

示例结果:

p
y
t
h
o
n

当我们使用next()迭代时,如果迭代了次数超过了迭代器中的元素个数就会引发 stopiteration ,由此我们可以利用 while 循环迭代,并不断捕捉迭代结束的异常完成for循环的迭代过程。

s = 'python'
it = iter(s)
while true:
  try:
    print(next(it))
  except stopiteration:
    break

p
y
t
h
o
n

手动构建迭代器

把一个类作为一个迭代器使用需要在类中实现两个方法 iter () 与 next () 。 iter () 方法返回一个特殊的迭代器对象, 这个迭代器对象实现了 next () 方法并通过 stopiteration 异常标识迭代的完成。 next () 方法(python 2 里是 next())会返回下一个迭代器对象。下面这个类构造了一个迭代器接受一个可迭代的数字,每次迭代都返回上一次迭代结果的平方,当迭代结果大于 9999999999999 时抛出异常。利用该类创建示例对数字2进行平方和迭代。

class it_square:
  def __init__(self, x):
    self.x = x

  def __next__(self):
    self.x = self.x ** 2
    if self.x > 9999999999999:
      raise stopiteration
    
    else:
      return self.x

  def __iter__(self):
    return self
 
it1 = it_square(2)
 
while true:
  try:
    print(it1.__next__())
  except stopiteration:
    break

示例结果

4
16
256
65536
4294967296

总结

以上所述是小编给大家介绍的python3之手动创建迭代器的实例代码,希望对大家有所帮助