해커를 꿈꾸는 개발자

파이썬 데이터 분석 / Numpy 배열 전체, 행별, 열별 sum, average 함수 사용

Numpy 배열 생성

randn 함수를 사용하여, 배열을 생성해보겠습니다.

randn 함수는 표준 정규 분포 ( N(0,1) 즉, 평균이 0이고 표준편차가 1인 분포) 에서 추출한 샘플을 반환하며, 음수의 값이 나올 수 있습니다. 

arr = np.random.randn(3,5)

행을 기준으로 연산

2차원 배열에서 행을 기준으로 연산할 경우에는 axis = 0 를 연산함수의 인자로 넣어 계산을 합니다. 

열을 기준으로 연산

2차원 배열에서 열을 기준으로 연산할 경우에는 axis = 1 을 연산함수의 인자로 넣어 계산을 합니다.

예제를 통해 이해하기

import numpy as np

arr = np.random.randn(3,5)
print(arr)
print("max : ", np.max(arr))

#2차원 배열의 모든 값의 합을 구합니다.
print("sum : ", np.sum(arr))


#2차원 배열의 행을 기준으로 평균을 구합니다.
#5개의 열값이 출력됩니다.
print("average : ", np.average(arr, axis=0))

#2차원 배열의 열을 기준으로 힙을 구합니다.
#3개의 행값이 출력됩니다.
print("sum : ", np.sum(arr, axis=1))