Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python – NumPy로 행렬 평균 구하기


행렬의 모든 요소에 대한 평균값을 구해야 할 때는 NumPy 패키지를 먼저 임포트한 뒤, 해당 패키지에서 제공하는 mean() 메서드를 사용하면 됩니다. NumPy는 수치 연산에 특화된 파이썬의 대표적인 라이브러리로, 복잡한 반복문 없이 단 한 줄의 코드로 통계 값을 손쉽게 계산할 수 있다는 장점이 있습니다.

예제 코드

아래는 행렬의 평균을 구하는 과정을 보여주는 예제입니다.

import numpy as np

my_matrix = np.matrix('[24, 41; 35, 25]')
print("The matrix is : ")
print(my_matrix)

my_result = my_matrix.mean()

print("The result is : ")
print(my_result)

실행 결과

The matrix is :
[[24 41]
[35 25]]
The result is :
31.25

코드 설명

  • 먼저 필요한 패키지(NumPy)를 실행 환경으로 임포트합니다.

  • np.matrix() 함수를 이용해 NumPy 패키지로 새로운 행렬을 생성합니다.

  • 생성된 행렬을 콘솔 화면에 출력합니다.

  • mean() 메서드를 호출하여 행렬 전체 요소의 평균값을 계산합니다.

  • 계산된 결과는 별도의 변수에 저장됩니다.

  • 마지막으로 결과가 담긴 변수를 콘솔에 출력하여 확인합니다.

참고로 위 예제의 행렬은 [[24, 41], [35, 25]] 형태로 구성되어 있으며, 네 요소의 합은 125이고 요소의 개수는 4이므로 평균값인 31.25가 도출됩니다. 이처럼 NumPy의 mean() 메서드를 활용하면 행렬 크기와 무관하게 평균 계산을 매우 간편하게 처리할 수 있습니다.