博客

python的iter函数怎么使用

  • Post category:Python

iter函数是Python中的一个内置函数，用于返回一个迭代器对象，它可以逐个地访问可迭代对象中的元素。

iter函数的语法如下所示：

iter(iterable[, sentinel])

其中，第一个参数iterable是可迭代对象，在使用iter函数的过程中，该参数是必须要提供的。而第二个参数sentinel，是可选的，它是一个标识，用于表示迭代的结束。如果不提供sentinel参数，则可迭代对象必须是一个可终止对象。

下面我们来详细介绍iter函数的使用方法：

  1. 使用iter函数创建迭代器对象

iter函数最常用的方式就是创建一个迭代器对象，该迭代器对象用于逐个访问可迭代对象中的元素。以下是一个示例代码，用于创建一个列表对象，并通过iter函数返回对应的迭代器对象：

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

在这个示例代码中，我们首先创建一个列表对象lst，然后通过调用iter函数，将lst转化为一个迭代器对象it，通过it对象的__next__()方法可以逐个地访问lst中的元素。

  1. 自定义可迭代对象

除了可以将现有的列表、元组等可迭代对象通过iter函数转化为迭代器对象之外，我们也可以自定义自己的可迭代对象。下面是一个示例代码，用于自定义一个可迭代对象，并将其转化为一个迭代器对象：

class MyRange:
    def __init__(self, n):
        self.i = 0
        self.n = n

    def __iter__(self):
        return self

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

mr = MyRange(5)
it = iter(mr)
print(next(it))  # output: 0
print(next(it))  # output: 1
print(next(it))  # output: 2
print(next(it))  # output: 3
print(next(it))  # output: 4

在这个示例代码中，我们首先定义了一个名为MyRange的类，该类继承自object类，并实现了__iter__和__next__方法。MyRange类的__iter__方法返回了对象本身，而__next__方法则定义了迭代器的访问规则，当self.i小于self.n时，需要返回对应的值，并递增self.i；否则，我们需要抛出StopIteration异常，表示停止迭代。

我们接着创建了一个MyRange类的对象mr，并通过iter函数将它转化为一个迭代器对象it。最后，我们通过调用it对象的__next__()方法，逐个地访问迭代器对象中的元素（0~4）。

以上是Python中iter函数的使用攻略，希望能够帮助到大家。

Tags: