Python

Python如何生成一个迭代器

文 / 大炳哥 来源 / 原创 阅读 / 9610 1年前

1. 什么是迭代器

迭代器是一个可以记住遍历位置的对象。可以遍历诸如列表,字典及字符串等序列对象

迭代器的特点:

  • 迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。
  • 迭代器只能往前不会后退。

2. 两个基本的方法:iter()next()

序列对象可以利用iter()直接创建迭代器,并通过next()获取迭代器中的下一个元素。

# 创建迭代器对象

S = 'PYTHON'
str_iter = iter(S)
print(type(str_iter))

# 利用for循环迭代

S = 'PYTHON'
str_iter = iter(S)
for word in str_iter:
    print(word)

# 利用next() 迭代

S = 'PYTHON'
str_iter = iter(S)
print(next(str_iter))
print(next(str_iter))
print(next(str_iter))
print(next(str_iter))
print(next(str_iter))
print(next(str_iter))

# 利用 while 循环迭代

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

S = 'PYTHON'
str_iter = iter(S)
while True:
    try:
        print(next(str_iter))
    except StopIteration:
        break

3. 手动构建迭代器

把一个类作为一个迭代器使用需要在类中实现两个方法 __iter__()__next__()

__iter__() 方法返回一个特殊的迭代器对象, 这个迭代器对象实现了 __next__() 方法并通过 StopIteration 异常标识迭代的完成。

__next__() 方法(Python 2 里是 next())会返回下一个迭代器对象。

下面这个类构造了一个迭代器接受一个可迭代的数字,每次迭代都返回上一次迭代结果的平方,当迭代结果大于9999999999999时抛出异常。利用该类创建示例对数字2进行平方和迭代。

class 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


# 直接调用
num = Square(2)

while True:
    try:
        print(num.__next__())
    except StopIteration:
        break

# 通过循环调用
for i in num:
    print(i)

# 通过next()调用
num = Square(2)
my_iter = iter(num)

print(next(my_iter))
print(next(my_iter))
print(next(my_iter))
print(next(my_iter))
print(next(my_iter))

# 通过while循环调用
num = Square(2)
my_iter = iter(num)

while True:
    try:
        print(next(my_iter))
    except StopIteration:
        break

12

评论

共0条评论
  • 这篇文章还没有收到评论,赶紧来抢沙发吧~

站点声明:本站转载作品版权归原作者及来源网站所有,原创内容作品版权归作者所有,任何内容转载、商业用途等均须联系原作者并注明来源。

© 2019 The Pride & Joy of an Artisan. · Designed By duoguyu · Powered By Django京ICP备19006898号-1
相关侵权、举报、投诉及建议等,请发E-mail:lemont0086@163.com

友情链接: 大炳哥的博客 逐梦个人博客 微微的博客 daixu 十年之约