Question

我想在a和b之间选择一个随机整数。

我知道3种方法。但是，他们的表现似乎非常违反直觉：

import timeit

t1 = timeit.timeit("n=random.randint(0, 2)", setup="import random", number=100000)
t2 = timeit.timeit("n=random.choice([0, 1, 2])", setup="import random", number=100000)
t3 = timeit.timeit("n=random.choice(ar)", setup="import random; ar = [0, 1, 2]", number=100000)

[print(t) for t in [t1, t2, t3]]

在我的机器上，这给出了：

0.29744589625620965
0.19716156798482648
0.17500512311108346

使用online interpreter，这会给出：

0.23830216699570883
0.16536146598809864
0.15081614299560897

请注意使用专用函数执行我正在做的事情的最直接版本（＃1）是如何 50％更差，这是预先定义数组的最奇怪版本（＃3）然后从中随机选择。

发生了什么事？

Answer 1

它只是实施细节。 randint委托给randrange，因此它有另一层函数调用开销，randrange经历了很多参数检查和其他问题。相比之下，choice是一个非常简单的单行。

以下是此次调用的代码路径randint，其中删除了注释和未执行的代码：

def randint(self, a, b):
    return self.randrange(a, b+1)

def randrange(self, start, stop=None, step=1, _int=int, _maxwidth=1L<<BPF):
    istart = _int(start)
    if istart != start:
        # not executed
    if stop is None:
        # not executed

    istop = _int(stop)
    if istop != stop:
        # not executed
    width = istop - istart
    if step == 1 and width > 0:
        if width >= _maxwidth:
            # not executed
        return _int(istart + _int(self.random()*width))

以下是代码路径choice：

def choice(self, seq):
    return seq[int(self.random() * len(seq))]

选择与randint的表现

1 个答案: