用Cython遍歷Numpy數組

Question

設a和b為兩個長度為1024的numpy.float數組，定義為

cdef numpy.ndarray a
cdef numpy.ndarray b

我注意到：

cdef int i
for i in range(1024):
    b[i] += a[i]

比：

b += a 

為什么？

我真的需要能夠手動循環遍歷數組。

Answer 1

如果您告訴Cython a和b的數據類型和維數，則差異會更小：

cdef numpy.ndarray[np.float64_t, ndim=1] a, b

盡管差異會更小，但您不會擊敗b += a因為這是使用NumPy的SIMD增強功能（該功能將取決於您的CPU是否支持SIMD）。