xrange 与 itertools.count Python 2.7

Question

我想运行一个从开始值到结束值的范围。它在低数字上工作正常，但是当它变得太大时会导致溢出错误，因为 int 太大而无法转换为 C Long。我正在使用 Python 2.7.3。

我在此处阅读了OverflowError Python int too large to convert to C long 关于使用itertools.count() 方法的内容，除了该方法与xrange 方法的工作方式不同，它通过步进而不是声明结束范围值。

itertools.count() 可以设置为像xrange() 一样工作吗？

print "Range start value"
start_value = raw_input('> ')
start_value = int(start_value)

print "Range end value"
end_value = raw_input('> ')
end_value = int(end_value)

for i in xrange(start_value, end_value):
    print hex(i)

Answer 1

您可以使用itertools.islice() 结束count：

from itertools import count, islice

for i in islice(count(start_value), end_value - start_value):

islice() 在 end_value - start_value 值被迭代后引发 StopIteration。

支持 1 以外的步长并将其全部放在一个函数中：

from itertools import count, islice

def irange(start, stop=None, step=1):
    if stop is None:
        start, stop = 0, start
    length = 0
    if step > 0 and start < stop:
        length = 1 + (stop - 1 - start) // step
    elif step < 0 and start > stop:
        length = 1 + (start - 1 - stop) // -step
    return islice(count(start, step), length)

然后像使用 range() 或 xrange() 一样使用 irange()，但现在可以使用 Python long 整数：

>>> import sys
>>> for i in irange(sys.maxint, sys.maxint + 10, 3):
...     print i
... 
9223372036854775807
9223372036854775810
9223372036854775813
9223372036854775816