NumPy의 행렬 열 합계

1. numpy.sum()함수를 사용하여 Python에서 행렬의 열 합계 찾기
2. numpy.einsum()함수를 사용하여 Python에서 행렬의 열 합계 찾기
3. numpy.dot()함수를 사용하여 Python에서 행렬의 열 합계 찾기

이 튜토리얼은 NumPy에서 열을 따라 요소의 합계를 찾는 방법을 소개합니다.

다음 행렬의 합을 계산합니다.

import numpy as np

a = np.arange(12).reshape(4, 3)
print(a)

출력:

[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [ 9 10 11]]

numpy.sum()함수를 사용하여 Python에서 행렬의 열 합계 찾기

sum()함수는 지정된 축에서 배열의 모든 요소의 합을 계산합니다. 축을 0으로 지정하면 행렬의 열에 대한 합계를 계산합니다.

다음 코드는이를 설명합니다.

import numpy as np

a = np.arange(12).reshape(4, 3)
s = np.sum(a, axis=0)
print(s)

출력:

[18 22 26]

이 방법은이 자습서에서 설명하는 모든 방법 중에서 가장 많이 사용되며 가장 빠릅니다.

numpy.einsum()함수를 사용하여 Python에서 행렬의 열 합계 찾기

einsum()은 NumPy에서 유용하지만 복잡한 함수입니다. 상황에 따라 다양한 방법으로 합계를 구할 수있어 설명하기 어렵다. 아래 그림과 같이이를 사용하여 행렬의 열 합계를 계산할 수 있습니다.

import numpy as np

a = np.arange(12).reshape(4, 3)
s = np.einsum("ij->j", a)
print(s)

출력:

[18 22 26]

ij->j는 배열 열의 합계를 계산해야 함을 지정하는 데 사용되는 함수의 첨자입니다.

numpy.dot()함수를 사용하여 Python에서 행렬의 열 합계 찾기

무관 한 방법이지만dot()함수의 방대한 사용을 이해하려면 여전히 알고 있어야합니다. 1 만 포함 된 단일 행 배열로 2 차원 배열의 내적을 계산하면이 행렬의 열 합계를 얻습니다.

다음 코드는이를 구현합니다.

import numpy as np

a = np.arange(12).reshape(4, 3)
s = np.dot(a.T, np.ones(a.shape[0]))
print(s)

출력:

[18. 22. 26.]