Numpyを使用して特定の行列の行と列の合計を見つける方法は？

この問題では、すべての行とすべての列の合計が別々に見つかります。合計を取得するためにsum（）関数を使用します。

アルゴリズム

Step 1: Import numpy.
Step 2: Create a numpy matrix of mxn dimension.
Step 3: Obtain the sum of all the rows.
Step 4: Obtain the sum of all the columns.

サンプルコード

import numpy as np

a = np.matrix('10 20; 30 40')
print("Our matrix: \n", a)

sum_of_rows = np.sum(a, axis = 0)
print("\nSum of all the rows: ", sum_of_rows)

sum_of_cols = np.sum(a, axis = 1)
print("\nSum of all the columns: \n", sum_of_cols)

出力

Our matrix:
 [[10 20]
 [30 40]]
Sum of all the rows:  [[40 60]]
Sum of all the columns:
 [[30]
 [70]]

説明

np.sum（）関数は、「軸」と呼ばれる追加の行列を取ります。 Axisは2つの値を取ります。 0または1のいずれか。axis=0の場合、sum（）関数に行のみを考慮するように指示します。 axis =1の場合、sum（）関数に列のみを考慮するように指示します。