any() 的性能

Question

这主要是一个学习 Python 的练习。我写了这个函数来测试一个数字是否是素数：

def p1(n):
    for d in xrange(2, int(math.sqrt(n)) + 1):
        if n % d == 0:
            return False
    return True

然后我意识到我可以使用 any() 轻松重写它：

def p2(n):
    return not any((n % d == 0) for d in xrange(2, int(math.sqrt(n)) + 1))

在性能方面，我期望 p2 比 p1 更快，或者至少和 p1 一样快，因为 any() 是内置的，但是对于一个大的素数，p2 会慢很多：

$ python -m timeit -n 100000 -s "import test" "test.p1(999983)"
100000 loops, best of 3: 60.2 usec per loop

$ python -m timeit -n 100000 -s "import test" "test.p2(999983)"
100000 loops, best of 3: 88.1 usec per loop

我在这里错误地使用了 any() 吗？有没有办法使用 any() 编写这个函数，这样它就可以迭代自己？

更新：更大素数的数字

$  python -m timeit -n 1000 -s "import test" "test.p1(9999999999971)"
1000 loops, best of 3: 181 msec per loop

$  python -m timeit -n 1000 -s "import test" "test.p2(9999999999971)"
1000 loops, best of 3: 261 msec per loop

Answer 1

性能差异很小，但它存在的原因是 any 与 for 循环相比，需要构建生成器表达式和额外的函数调用。但是，两者具有相同的行为（快捷方式评估）。

随着输入大小的增加，差异不会减少（我错了），因为您使用的是生成器表达式，并且迭代它需要在其上调用一个方法 (.next()) 和一个额外的堆栈框架。 any 当然是在幕后做的。

for 循环正在迭代 xrange 对象。 any 正在迭代生成器表达式，而生成器表达式本身正在迭代 xrange 对象。

无论哪种方式，请使用生成最易读/可维护的代码。选择其中一个对您正在编写的任何程序的性能影响很小（如果有的话）。