快速求和数组

Question

我正在尝试使用

对像向量之类的数组求和

import numpy as np

def a_sum(array_list):
    array_sum = array_list[0]
    for i in range(len(array_list) - 1):
        array_sum = np.add(array_sum, array_list[i + 1])
    return array_sum

这个函数被多次调用，所以我需要以某种方式优化它。有办法吗？

示例输入输出：

In []: a_sum([[1,2],[3,4], [5,6]])
Out[]: array([ 9, 12])

Answer 1

您可以使用np.sum()。

import numpy as np

def a_sum(array_list):
    return np.sum(array_list, axis=0)

你会看到直接在你的代码中直接使用numpy数组的最大改进，否则numpy将总是将python列表转换为其数据类型，浪费时间。

Answer 2

只需使用np.sum()：

np.sum(array_list)

Answer 3

你应该使用np.sum

np.sum(array_list)