Python:如何在不创建范围的情况下计算范围的长度？

Question

我需要计算一个范围的长度，但最好不要创建范围(希望会更快并且使用更少的内存。这很重要，因为这个函数会被调用很多)。该长度用于设置扩展切片。

目前我已经尝试过:

int_div = lambda n, d: (n + d // 2) // d

def range_len(start, stop, step):
    return int_div(stop - start, step)

但在某些情况下，例如 range_len(9, 100, 3) 它在正确答案为 31 时给出 30。我觉得这应该很简单，我做错了什么？

Answer 1

您可以使用这个公式:(end - start - 1)//step + 1

def calc_length(start, end, step):
    return (end - start - 1) // step + 1

for i in range(start, end):
    calculated = calc_length(start, i, step)
    empirical = len(range(start, i, step))

    assert calculated == empirical, "{} {}".format(calculated, empirical)