如何使用 Numpy 找到给定矩阵的行和列的总和?
pythonnumpyserver side programmingprogramming更新于 2024/9/21 20:28:17
在这个问题中,我们将分别找到所有行和所有列的总和。我们将使用 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() 函数仅考虑列。
相关文章
有用资源
python 参考教程 - 该教程包含有关 python 的更多信息:https://www.cainiaomax.com/python/

