Cython 优化 numpy 数组求和的关键部分

Question

令 L 是一个列表 L = [A_1, A_2, ..., A_n]，每个 A_i 都是长度为 1024 的 numpy.int32 数组。

（大多数时候 1000

经过一些分析，我发现最耗时的操作是求和：

def summation():
    # L is a global variable, modified outside of this function
    b = numpy.zeros(1024, numpy.int32)
    for a in L:
        b += a
    return b

PS：我不认为我可以定义大小为 1024 x n 的二维数组，因为 n 不固定：一些元素被动态删除/添加到 L，所以 len(L) = n 在期间可以在 1000 和 4000 之间变化运行时间。

我可以通过使用 Cython 获得显着的改进吗？ 如果是这样，我应该如何对这个小功能进行cython-recode（我不应该添加一些cdef打字吗？）

或者你能看到一些其他可能的改进吗？

Answer 1

这是 Cython 代码，确保 L 中的每个数组都是 C_CONTIGUOUS：

import cython
import numpy as np
cimport numpy as np

@cython.boundscheck(False)
@cython.wraparound(False)
def sum_list(list a):
    cdef int* x
    cdef int* b
    cdef int i, j
    cdef int count
    count = len(a[0])
    res = np.zeros_like(a[0])
    b = <int *>((<np.ndarray>res).data)
    for j in range(len(a)):
        x = <int *>((<np.ndarray>a[j]).data)
        for i in range(count):
            b[i] += x[i]
    return res

一台我的 PC 大约快 4 倍。