python - Python中的循环范围

Question

如何在 Python 中实现圆形范围对象？

例如

令 S 是一个以 2^3 为模的圆形空间(范围 [0, 2^3))。我想生成一个像这样的范围对象:

crange(3, 7, 2 ** 3) => a range object [3, 4, 5, 6]
crange(7, 3, 2 ** 3) => a range object [7, 0, 1, 2]

我试过这个:

def crange(start, stop, modulo):
    if start > stop:
        return range(start, modulo) or range(stop)
    else:
        return range(start, stop)

但我不能输入 bigint 到 crange 例如crange(8, 2, 2 ** 160).

OverflowError: Python int too large to convert to C ssize_t

Answer 1

试试这个:

def crange(start, stop, modulo):
    result = []
    index = start
    while index != stop:
        result.append(index)
        index = (index + 1) % modulo
    return result

如果您知道您的列表可能太长，您可以使用生成器来生成必需序列:

def crange(start, stop, modulo):
    index = start
    while index != stop:
        yield index
        index = (index + 1) % modulo