python中的for循环到底做了些啥

本文探讨了Python中for循环的本质,通过实例解析了for ... in ...语句如何灵活地遍历列表、range对象和enumerate对象。介绍了迭代器协议,包括__iter__和__next__方法,并揭示了range函数如何实现迭代数据与对象的分离,增强了灵活性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

for循环的本质

我们常常会写如下的代码

a = [1,2,3]
for item in a:
    print(item)

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

for i, item in enumerate(a):
    print(i, item)

第一部分代码例子很好理解,循环遍历列表 a 里面的元素。 python 的 for … in … 语义非常清晰。
第二部分代码例子和第一部分比较起来,似乎 range(10) 和列表等价。
第三部分代码例子和第一部分比较起来, enumerate(a) 每次迭代会返回两个元素,第一个 i 代表序号。
究竟是不是这样呢。 python 这几个关键字是怎么让其在迭代用法这么灵活呢(想想java 等其他语言吧)。
有部分读者可能有了解过。迭代器协议
迭代器协议:

  • 实现 __iter__ 方法, 该方法返回迭代器对象。
  • 实现 __next__ 方法,该方法返回迭代的值,抛出 StopIteration 表示停止迭代。

可能还是云里雾里的,没关系,我们直接看一个例子。

class A:
    def __init__(self, end):
        self.i = -1
        self.end = end

    def __iter__(self):
        return self

    def __next__(self):
        self.i += 1
        if self.i < self.end:
            return self.i
        else:
            raise StopIteration()

for i in A(10):
    print(i)

这个时候 我们可以发现 A(10) 的效果和 range(10) 一毛一样! 打印从 0 - 9 。
那么 range 函数是不是这样做的呢。 我们来看看。 在看之前我们需要了解一个知识点。

  • 工具 dir 函数。 内置 dir 函数能获取对象的属性和方法描述。
print(dir(range(10)))

# 输出结果
['__bool__', '__class__', '__contains__', '__delattr__', '__dir__', '__doc__', '__eq__', '__format__', '__ge__', '__getattribute__', '__getitem__', '__gt__', '__hash__', '__init__', '__init_subclass__', '__iter__', '__le__', '__len__', '__lt__', '__ne__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__reversed__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'count', 'index', 'start', 'step', 'stop']

我们可以看到, range(10) 确实有一个 __iter__ 方法,但是它并没有__next__ 方法。这是怎么回事呢。

print(dir(range(10).__iter__()))

# 输出结果
['__class__', '__delattr__', '__dir__', '__doc__', '__eq__', '__format__', '__ge__', '__getattribute__', '__gt__', '__hash__', '__init__', '__init_subclass__', '__iter__', '__le__', '__length_hint__', '__lt__', '__ne__', '__new__', '__next__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__setstate__', '__sizeof__', '__str__', '__subclasshook__']

奥,原来__next__方法在这。 range(10) 的实现将迭代数据和对象本身进行了分离。 这样你对上文的描述是不是理解的更深了一些。 这也就是__iter__方法的用处。很多书本可能没提到这点。 甚至直接搞混淆的,说返回其本身。__iter__ 返回含有__next__方法的对象。而不想分离的时候就返回对象本身。这种设计是不是更灵活了呢。

  • 动手题
    1.自己试试探索一下文章开头的其余两个代码例子吧。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值