容器的迭代工具

Question

考虑以下交互式示例

>>> l=imap(str,xrange(1,4))
>>> list(l)
['1', '2', '3']
>>> list(l)
[]

有谁知道某处是否已经有一个带有 imap 版本（和其他 itertools 函数）的实现，以便第二次执行 list(l) 得到与第一次相同的结果。而且我不想要常规映射，因为如果您使用更大的范围，在内存中构建整个输出可能会浪费内存。

我想要一些基本上可以做类似事情的东西

class cmap:
    def __init__(self, function, *iterators):
        self._function = function
        self._iterators = iterators

    def __iter__(self):
        return itertools.imap(self._function, *self._iterators)

    def __len__(self):
        return min( map(len, self._iterators) )

但如果有人已经这样做了，那么手动为所有 itertools 执行此操作将是浪费时间。

ps。 你认为容器比迭代器更禅宗吗，因为对于像

这样的迭代器

for i in iterator:
    do something

当您明确需要删除元素的容器时，隐式清空迭代器。

Answer 1

您不必为每种类型的容器构建这样的对象。基本上，你有以下几点：

mkimap = lambda: imap(str,xrange(1,4))
list(mkimap())
list(mkimap())

现在你需要一个漂亮的包装对象来防止“丑陋”的函数调用。这可以这样工作：

class MultiIter(object):
    def __init__(self, f, *a, **k):
        if a or k:
            self.create = lambda: f(*a, **k)
        else: # optimize
            self.create = f
    def __iter__(self):
        return self.create()

l = MultiIter(lambda: imap(str, xrange(1,4)))
# or
l = MultiIter(imap, str, xrange(1,4))
# or even
@MultiIter
def l():
    return imap(str, xrange(1,4))

# and then
print list(l)
print list(l)

（未经测试，希望它有效，但你应该明白）

对于您的第二个问题：迭代器和容器都有其用途。你应该选择最适合你需要的东西。

Answer 2

您可能正在寻找itertools.tee()

Answer 3

迭代器是我最喜欢的话题；）

from itertools import imap

class imap2(object):
    def __init__(self, f, *args):
        self.g = imap(f,*args)
        self.lst = []
        self.done = False

    def __iter__(self):
        while True:
            try: # try to get something from g
                x = next(self.g)
            except StopIteration:
                if self.done:
                    # give the old values
                    for x in self.lst:
                        yield x
                else:
                    # g was consumed for the first time
                    self.done = True
                return
            else:
                self.lst.append(x)
                yield x




l=imap2(str,xrange(1,4))
print list(l)
print list(l)