이 문제에서는 모든 행과 모든 열의 합을 개별적으로 찾습니다. 합계를 구하기 위해 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는 두 개의 값을 취합니다. 0 또는 1입니다. axis=0이면 sum() 함수에 행만 고려하도록 지시합니다. axis =1이면 sum() 함수에 열만 고려하도록 지시합니다.