新彩天欢迎您!
幻海优品

Python NumPy ufunc sum 求和

1、sum

sum和add有什么区别?

在两个参数之间进行add,而sum发生在n个元素上。

例如:

将arr1中的值添加到arr2中的值:

import numpy as nparr1 = np.array([1, 2, 3])arr2 = np.array([1, 2, 3])newarr = np.add(arr1, arr2)print(newarr)

返回[2 4 6]

例如:

将arr1中的值与arr2中的值相加:

import numpy as nparr1 = np.array([1, 2, 3])arr2 = np.array([1, 2, 3])newarr = np.sum([arr1, arr2])print(newarr)

返回:12

2、对轴求和

如果指定axis = 1,则NumPy将对每个数组中的数字求和。

例如:

在第一个轴上以以下数组进行求和:

import numpy as nparr1 = np.array([1, 2, 3])arr2 = np.array([1, 2, 3])newarr = np.sum([arr1, arr2], axis=1)print(newarr)

返回:[6 6]

3、cumsum累计和

累积和表示将数组中的元素部分相加。

例如,[1、2、3、4]的部分和是[1、1+2、1+2+3、1+2+3+4]=[1、3、6、10]。

使用cumsum()函数执行部分求和。

例如:

在以下数组中执行累积求和:

import numpy as nparr = np.array([1, 2, 3])newarr = np.cumsum(arr)print(newarr)

返回:[1 3 6]

免责声明:以上内容(如有图片或视频亦包括在内)有转载其他网站资源,如有侵权请联系删除