«

Python中怎么创建迭代器

时间:2024-6-8 09:49     作者:韩俊     分类: Python


本篇内容主要讲解“Python中怎么创建迭代器”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“Python中怎么创建迭代器”吧!

什么是迭代器

什么是

python
迭代器呢? 举一个最简单的例子:

list1 = ["pdudo","juejin","hello"]
tuple1 = ("pdudo","juejin","hello")
dict1 = {"name":"pdudo"}

for i in list1:
    print(i)
for i in tuple1:
    print(i)
for i in dict1:
    print(i)

这就是

python
迭代器,好了,讲完了,手工。

是的,使用

for...in
的方式,底层都是使用的是迭代器,你是不是之前写的时候,从来没有好奇过,为什么遍历不同的数据类型,都可以使用
for...in
通用呢?

为什么需要迭代器

只要符合

python
迭代器条件的,都可以使用
for...in
来遍历元素,即: 使用相同的代码,遍历不同的数据容器。 我认为这是根本原因。

如果上述描述还不清晰的话,我们可以使用

c
python
来遍历一下数组 和 字符串,就能清晰的了解了。

如上代码,是

c
语言遍历数组
"pdudo","hello","juejin"
和字符串
pdudohellojuejin
,我们需要自己写遍历条件的临界值。

而如何使用

python
来写呢? 来看下呢。

list1 = ["pdudo","hello","juejin"]
char1 = "pdudohellojuejin"

for v in list1:
    print(v)

for c in char1:
    print(c)

只需要定义数组和字符串,而后使用

for...in
便结束了。

我想,如上例子,就足以证明为什么要使用迭代器的原因了,因为真的很爽。

迭代器是如何工作的

在经历了前2个段落的铺垫,我猜你肯定很想知道迭代器是如何工作的吧?现在它来了。

在使用

for...in
语句时,它会调用
inter()
对象,该函数会返回一个迭代器对象。该对象又定义了
__next__()
方法,该方法一次返回一个容器元素,当没有更多元素可以返回的时候,会抛一个
StopIteration
异常来表明
for
终止循环。

是不是还是不懂?没关系,我们再写一个案例来说明一下。

list1 = ["pdudo","hello","juejin"]

it = iter(list1)
print(next(it))
print(next(it))
print(next(it))
print(next(it))

如上代码,定义了一个列表,其值为: "pdudo","hello","juejin",而后调用

iter
方法,它将返回一个迭代器,而后调用
next
方法来返回下一个元素,但是我们定义的列表长度为3,而调用了4次
next
方法,可见,最后一次会抛异常。

我们执行后,效果如下:

可见,和我们上述猜想的一致,在

for...in
语句中,也是调用
inter()
容器对象,使用
__next__
返回后续可迭代的对象,如此类推,直至遇到异常
StopIteration
,循环结束。

好了,知道迭代器是如何工作了吧? 那么,我们再抛出一个问题,看你能否接住呢? 如何判断一个数据类型是能够被迭代的呢?

如何创建一个迭代器

我们已经学会了如何使用迭代器,以及知晓了迭代器是如何工作的,本段落将介绍如何创建一个迭代器,在看这个之前,我们思考一个问题,如下代码是否会报错呢?

a = 9527
for i in a:
    print(i)

我们使用

for...in
来遍历一个
int
类型的数据。

如上代码,当然会报错,借此引出我们的知识点:什么样的数据类型才能被迭代呢?

是这样的,能否被迭代,取决于该方法是否有

__iter__
方法。

可以看下如下例子,我们自定义了一个迭代器,用于倒叙输出数据。

#!/bin/env python

class flashBack:
    def __init__(self,list1):
        self.data = list1
        self.index = len(list1)-1

    def __iter__(self):
        return self

    def __next__(self):
        if self.index < 0:
            raise StopIteration

        val = self.data[self.index]
        self.index = self.index - 1
        return val

def main():
    list1 = [1,2,3,4,5]
    tuble1 = ("pdudo","juejin","hello")

    for i in flashBack(list1):
        print(i)

    for i in flashBack(tuble1):
        print(i)

if __name__ == '__main__':
    main()

执行后,结果为:

可见,创建一个迭代器,至少需要

__iter__
方法 和 有
__next__
方法。

好了,有了这个基础案例,我们来写一个链表?

#!/bin/env python

class Node:
    def __init__(self,val):
        self.val = val
        self.nextNode = None

class Lists:
    def __init__(self,currentNodes):
        self.currentNode = currentNodes

    def __iter__(self):
        return self

    def __next__(self):
        if self.currentNode is None:
            raise StopIteration

        v = self.currentNode.val
        self.currentNode = self.currentNode.nextNode
        return v

def main() :
    node1 = Node(1)
    node2 = Node(2)
    node3 = Node(3)
    node1.nextNode = node2
    node2.nextNode = node3

    for i in Lists(node1):
        print(i)

if __name__ == '__main__':
    main()

如上代码,我们先创建节点

Node
,它有2个值,
val
是记录的值,而
nextNode
是记录下一个
Node
的指针,而后定义了类
Lists
,调用时候,需要传入一个
Node
,它会将
currentNodes
来记录当前的
Node
重点看
__next__
,当当前节点为空的时候,则返回
StopIteration
告知
for
迭代器结束了,否则的话,取出当前节点的
val
并且返回,且将其下滑到下一个节点。

如上代码,运行后,结果如下:

标签: python

热门推荐