python - numpy数组求和，哪种方法比较好用？

Question

以下3种方法有什么区别，哪种更好用？

import numpy as np

a = np.array([1, 2])
b = np.array([2, 1])

sum(a*b) # mtd 1
(a*b).sum() # mtd 2
np.sum(a*b) # mtd 3

Answer 1

最好的是np.dot(a, b)这恰好适用于您正在执行的操作:将 a 的条目乘以 b 的条目，然后将它们相加。

至于np.sum(array)对比array.sum() ，这是表达同一事物的两种方式，提供这两种形式是为了方便。

正如 ajcr 所说，sum(a*b)是次优的，因为它是一个普通的 Python 方法，而不是 NumPy 方法。 Python 的 sum必须处理 Python 列表中存在的各种事物。 NumPy 数组是同类的，包含相同类型的对象，因此它的方法利用了这一点。