numpy vs list comprehension，哪个更快？

Question

我跑了一个简单的速度测试，比较numpy和Python列表的理解，显然列表理解更快。那是对的吗？

import sys, numpy 
from datetime import datetime 

def numpysum(n): 
    a = numpy.arange(n) ** 2 
    b = numpy.arange(n) ** 3 
    return a + b 

def pythonsum(n): 
    a = [i ** 2 for i in range(n)] 
    b = [i ** 3 for i in range(n)] 
    return [a[i] + b[i] for i in range(n)] 

size = 10 
start = datetime.now() 
c1 = pythonsum(size) 
delta1 = datetime.now() - start 

start = datetime.now() 
c2 = numpysum(size) 
delta2 = datetime.now() - start 

print c1 
print c2 

print delta1 
print delta2 

Answer 1

我想你可能要考虑不同的测试参数：与NumPy的VS列表理解时序

In [39]: %timeit pythonsum(10) 
100000 loops, best of 3: 8.41 us per loop 

In [40]: %timeit pythonsum(100) 
10000 loops, best of 3: 51.9 us per loop 

In [41]: %timeit pythonsum(1000) 
1000 loops, best of 3: 451 us per loop 

In [42]: %timeit pythonsum(10000) 
100 loops, best of 3: 17.9 ms per loop 

In [43]: %timeit numpysum(10) 
100000 loops, best of 3: 13.4 us per loop 

In [44]: %timeit numpysum(100) 
100000 loops, best of 3: 17 us per loop 

In [45]: %timeit numpysum(1000) 
10000 loops, best of 3: 50.3 us per loop 

In [46]: %timeit numpysum(10000) 
1000 loops, best of 3: 385 us per loop 

比：

10：0.6X

100：3.1 x

1000：9x

10000：46X

因此，numpy的是大N快得多。

Answer 2

您的size太小。我再次尝试size=1000000，numpy的表现理解度超过了9x。

我猜numpy的具有更高的设置开销，但在一般的非平凡输入尺寸（10绝对是微不足道的），你可以期望它至少是一样快的列表理解，在大多数情况下多更快。