Python numpy通过sum函数计算数组中元素的和
在NumPy中，numpy.sum()函数可以计算数组中元素的和。这个函数可以接受一个参数axis，用于指定计算的轴。如果不指定axis参数，则默认计算整个数组的和。
以下是一些使用numpy.sum()函数计算数组元素和的示例：
import numpy as np # 计算整个数组的元素和 arr = np.array([1, 2, 3, 4, 5]) sum_all = np.sum(arr) print("整个数组的元素和:", sum_all) # 计算二维数组按行的元素和 arr_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) sum_row = np.sum(arr_2d, axis=1) print("按行的元素和:", sum_row) # 计算二维数组按列的元素和 sum_col = np.sum(arr_2d, axis=0) print("按列的元素和:", sum_col)
输出：
整个数组的元素和: 15 按行的元素和: [ 6 15 24] 按列的元素和: [12 15 18]
在这个示例中，我们首先使用numpy.sum()函数计算了一个一维数组的元素和；然后，使用numpy.sum()函数计算了一个二维数组按行的元素和和按列的元素和。在计算二维数组的元素和时，我们通过指定axis参数来控制计算的轴。当axis参数等于1时，计算按行的元素和；当axis参数等于0时，计算按列的元素和。
相关文章