[python] NumPy에서 2차원 배열에서 각 열의 평균을 구하는 방법

개요

2차원 배열을 생성한 후에, numpy를 이용하여 각 열의 평균을 구하는 방법을 알아 보겠습니다.

numpy mean()

NumPy.array의 평균을 구할때는 mean() 함수를 이용하면 됩니다.

import numpy as np

# 2차원 배열 생성
arr2d = np.array([[1, 2, 3], 
                [4, 5, 6], 
                [7, 8, 9]])

print(arr2d)
# [[1 2 3]
#  [4 5 6]
#  [7 8 9]]

# 열의 평균 계산 - axis=0
col_means = arr2d.mean(axis = 0)

print(col_means)
# [4. 5. 6.]

 

위 예제 코드와 같이 "axis = 0"으로 mean() 함수를 호출하면 각 열의 평균을 구할 수 있습니다.

"axis = 1"로 mean() 함수를 호출하면 각 행의 평균을 구할 수 있습니다.

import numpy as np

# 2차원 배열 생성
arr2d = np.array([[1, 2, 3], 
                [4, 5, 6], 
                [7, 8, 9]])

# 행의 평균 계산 - axis=1
row_means = arr2d.mean(axis = 1)
print(row_means)
# [2. 5. 8.]