当前位置：首页 > article >正文

python iter函数介绍

article 2025/3/1 15:53:29

1. 功能

iter() 函数用于返回一个对象的迭代器。迭代器是遵循迭代协议的对象，能够逐一访问序列中的元素而无需暴露底层实现。

2. 语法

iter(object[, sentinel])

object：必须是一个可迭代对象（如列表、元组、字符串、字典、集合）或者实现了 __iter__() 方法。
sentinel（可选）：一个标记值，仅在对象是一个可调用对象时使用。

3. 用法

a. 可迭代对象创建迭代器

如果 object 是一个可迭代对象，iter() 会返回该对象的迭代器。

lst = [1, 2, 3, 4]
it = iter(lst)

print(next(it))  # 输出: 1
print(next(it))  # 输出: 2
print(next(it))  # 输出: 3
print(next(it))  # 输出: 4
# 再次调用 next(it) 会抛出 StopIteration 异常

b. 可调用对象与哨兵值

如果 object 是一个可调用对象，且提供了 sentinel 参数，则 iter() 会持续调用该对象，直到返回值等于哨兵值为止。

def counter():
    i = 0
    while True:
        i += 1
        yield i

c = counter()
it = iter(c, 5)  # 当返回值等于 5 时停止迭代

for num in it:
    print(num)  # 输出: 1, 2, 3, 4

4. 与自定义类配合使用

一个对象要支持 iter()，必须实现以下两种方法之一：

__iter__() 方法：返回自身迭代器。
__getitem__() 方法：支持从索引 0 开始的序列访问。

实现 `iter()` 的自定义类

class MyRange:
    def __init__(self, start, end):
        self.current = start
        self.end = end

    def __iter__(self):
        return self

    def __next__(self):
        if self.current >= self.end:
            raise StopIteration
        value = self.current
        self.current += 1
        return value

# 使用 iter
my_range = MyRange(1, 5)
it = iter(my_range)

for num in it:
    print(num)  # 输出: 1, 2, 3, 4