向xrange()添加任意元素？

Question

在Python中，在迭代时使用xrange()而不是range更节省内存。

我遇到的问题是，我想迭代一个大列表--因此我需要使用xrange()，然后检查一个任意的元素。

对于range()，它很简单：x = range(...) + [arbitrary element]。

但是使用xrange()，似乎没有比这更干净的解决方案了：

for i in xrange(...):
    if foo(i):
        ...
if foo(arbitrary element):
        ...

对清洁的解决方案有什么建议吗？有没有办法将任意元素“附加”到生成器中？

Answer 1

itertools.chain允许您从多个迭代器中生成一个组合迭代器，而不需要将它们连接起来(这样就没有昂贵的临时迭代器)：

from itertools import chain

# Must wrap arbitrary element in one-element tuple (or list)
for i in chain(xrange(...), (arbitrary_element,)):
    if foo(i):
        ...

Answer 2

我建议将arbitrary_element签入循环之外，但是如果您想使它成为循环的一部分，则可以使用itertools.chain。

for i in itertools.chain(xrange(...), [arbitrary_element]):
    ...