NumPy 求和

求和

求和与加法的区别是什么?

加法是在两个参数之间进行的,而求和是在 n 个元素上进行的。

NumPy 使用 sum() 函数来进行求和。

实例

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

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

结果:

  1. [2 4 6]
实例

将 arr1 中的值与 arr2 中的值全部加起来求和:

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

结果:

  1. 12

对轴求和

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

实例

在第 1 个轴上的以下数组中执行求和:

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

结果:

  1. [6 6]

累加求和

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

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

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

实例

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

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

结果:

  1. [1 3 6]

分类导航