Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

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는 두 개의 값을 취합니다. 0 또는 1입니다. axis=0이면 sum() 함수에 행만 고려하도록 지시합니다. axis =1이면 sum() 함수에 열만 고려하도록 지시합니다.