如何使用 Numpy 找到给定矩阵的行和列的总和?
pythonnumpyserver side programmingprogramming
在这个问题中，我们将分别找到所有行和所有列的总和。我们将使用 sum() 函数来获取总和。
算法
步骤 1:导入 numpy。 步骤 2:创建一个 mxn 维的 numpy 矩阵。 步骤 3:获取所有行的总和。 步骤 4:获取所有列的总和。
示例代码
import numpy as np
a = np.matrix('10 20; 30 40')
print("我们的矩阵:\n", a)
sum_of_rows = np.sum(a, axis = 0)
print("\n所有行的总和:", sum_of_rows)
sum_of_cols = np.sum(a, axis = 1)
print("\n所有列的总和:", sum_of_cols)
输出
我们的矩阵: [[10 20] [30 40]] 所有行的总和: [[40 60]] 所有列的总和: [[30] [70]]
解释
np.sum() 函数采用称为"axis"的附加矩阵。axis 接受两个值。0 或 1。如果 axis=0，它会告诉 sum() 函数仅考虑行。如果 axis = 1，它会告诉 sum() 函数仅考虑列。
