python - 向 xrange() 添加任意元素？

Question

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

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

使用 range()，很简单:x = range(...) + [任意元素]。

但是对于 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):
        ...