如何使用Numpy找到给定矩阵的行和列的和？

在这个问题中，我们将分别找到所有行和所有列的和。我们将使用sum()函数获取总和。

阅读更多：Python 教程

算法

步骤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所有列的总和：\n", sum_of_cols)

输出

我们的矩阵：
 [[10 20]
 [30 40]]
所有行的总和： [[40 60]]
所有列的总和：
 [[30]
 [70]]

说明

np.sum()函数接受一个名为’axis’的附加矩阵。Axis有两个值。要么是0或1。如果axis=0，则它告诉sum()函数只考虑行。如果axis=1，则它告诉sum()函数只考虑列。