检测构建独立迭代器的最便宜方法

Question

假设我正在编写一个接受可迭代对象的函数，并且我的函数希望不知道该可迭代对象是否实际上是迭代器。

（这是一种常见的情况，对吧？我认为基本上所有的 itertools 函数都是这样编写的。接受一个可迭代对象，返回一个迭代器。）

例如，如果我调用 itertools.tee(•, 2)< /code> 在一个对象上，而它恰好还不是一个迭代器，这可能意味着只需对其调用两次 iter 以获得两个独立的迭代器会更便宜。 itertools 函数是否足够聪明，能够知道这一点，如果没有，避免这种方式不必要的成本的最佳方法是什么？

Answer 1

观察：

>>> def foo(x):
...     return x.__iter__() # or return iter(x)
...
>>> l = [0, 1]
>>> it = l.__iter__()
>>> it
<list_iterator object at 0x00000190F59C3640>
>>> print(foo(l), foo(it))
<list_iterator object at 0x00000190F5980AF0> <list_iterator object at 0x00000190F59C3640>

因此您无需担心函数的参数是可迭代的还是已经是迭代器了。您可以对已经是迭代器的对象调用方法 __iter__ ，在这种情况下它只返回 self 。这不是一个昂贵的调用，并且比您可以用来测试它是否是迭代器的任何事情都要便宜，例如它是否有 __next__ 方法（以及如果没有的话，无论如何都必须调用 __iter__ ）。

更新

现在，我们发现，自从调用 <在前者上调用两次 iter 会给你两个不同的迭代器，而在后者上调用两次 iter 则不会。例如，itertools.tee 需要一个可迭代的对象。如果你向它传递一个实现 __iter__ 并返回 'self的迭代器，它显然会起作用，因为tee` 不需要两个独立的迭代器来发挥其魔力。

但是，如果您正在编写一个传递一个可迭代的迭代器，该迭代器是通过在传递的迭代器上内部使用两个或多个迭代器来实现的，那么您真正想要的是什么测试的是正在传递的内容是否支持多个、并发、独立的迭代，无论它是迭代器还是普通的迭代器：

def my_iterator(iterable):
     it1 = iter(iterable)
     it2 = iter(iterable)
     if it1 is it2:
          raise ValueError('The passed iterable does not support multiple, concurrent, independent iterations.')
     ...

class Foo:
     def __init__(self, lst):
          self.lst = lst

     def __iter__(self):
          self.idx = 0
          return self

     def __next__(self):
          if self.idx < len(self.lst):
               value = self.lst[self.idx]
               self.idx += 1
               return value
          raise StopIteration()

f = Foo("abcd")
for x in f:
     print(x)

my_iterator(f)

打印：

a
b
c
d
Traceback (most recent call last):
  File "C:\Booboo\test\test.py", line 26, in <module>
    my_iterator(f)
  File "C:\Booboo\test\test.py", line 5, in my_iterator
    raise ValueError('The passed iterable does not support multiple, concurrent, independent iterations.')
ValueError: The passed iterable does not support multiple, concurrent, independent iterations.

The writer of 原件，已通过迭代器必须以支持多个、并发、独立迭代的方式编写。