[파이썬 numpy] 배열의 평균 (axis 방향별)

[파이썬 numpy] 

배열의 평균 (axis 방향별)

배열의 평균을 구할때는 mean 메소드를 사용합니다.

먼저 배열을 하나 정의합시다.

>>> import numpy as np

>>> A=np.array([1,2,3,4,5])

배열의 평균

평균은 아래와 같이 두 가지 방법으로 구할 수 있습니다. 넘파이 함수를 사용해도 되고, 정의한 배열에 메소드를 적용해도 됩니다. 

>>> np.mean(A)

3.0

>>> A.mean()

3.0

배열의 axis 에 따른 평균

2차원 배열을 하나 정의합시다.

>>> A=np.arange(6).reshape(2,3)

>>> A

array([[0, 1, 2],

       [3, 4, 5]])

axis=0 방향의 평균을 구해봅시다. 행방향 평균입니다. 

>>> A.mean(axis=0)

array([1.5, 2.5, 3.5])

axis=1 방향의 평균을 구해봅시다. 열방향 평균입니다. 

A.mean(axis=1)

array([1., 4.])