도구 - 넘파이(NumPy)

*넘파이(NumPy)는 파이썬의 과학 컴퓨팅을 위한 기본 라이브러리입니다. 넘파이의 핵심은 강력한 N-차원 배열 객체입니다. 또한 선형 대수, 푸리에(Fourier) 변환, 유사 난수 생성과 같은 유용한 함수들도 제공합니다."

구글 코랩에서 실행하기

배열 생성

numpy를 임포트해 보죠. 대부분의 사람들이 np로 알리아싱하여 임포트합니다:

np.zeros

zeros 함수는 0으로 채워진 배열을 만듭니다:

2D 배열(즉, 행렬)을 만들려면 원하는 행과 열의 크기를 튜플로 전달합니다. 예를 들어 다음은 $3 \times 4$ 크기의 행렬입니다:

용어

N-차원 배열

임의의 랭크 수를 가진 N-차원 배열을 만들 수 있습니다. 예를 들어, 다음은 크기가 (2,3,4)인 3D 배열(랭크=3)입니다:

배열 타입

넘파이 배열의 타입은 ndarray입니다:

np.ones

ndarray를 만들 수 있는 넘파이 함수가 많습니다.

다음은 1로 채워진 $3 \times 4$ 크기의 행렬입니다:

np.full

주어진 값으로 지정된 크기의 배열을 초기화합니다. 다음은 π로 채워진 $3 \times 4$ 크기의 행렬입니다.

np.empty

초기화되지 않은 $2 \times 3$ 크기의 배열을 만듭니다(배열의 내용은 예측이 불가능하며 메모리 상황에 따라 달라집니다):

np.array

array 함수는 파이썬 리스트를 사용하여 ndarray를 초기화합니다:

np.arange

파이썬의 기본 range 함수와 비슷한 넘파이 arange 함수를 사용하여 ndarray를 만들 수 있습니다:

부동 소수도 가능합니다:

파이썬의 기본 range 함수처럼 건너 뛰는 정도를 지정할 수 있습니다:

부동 소수를 사용하면 원소의 개수가 일정하지 않을 수 있습니다. 예를 들면 다음과 같습니다:

np.linspace

이런 이유로 부동 소수를 사용할 땐 arange 대신에 linspace 함수를 사용하는 것이 좋습니다. linspace 함수는 지정된 개수만큼 두 값 사이를 나눈 배열을 반환합니다(arange와는 다르게 최댓값이 포함됩니다):

np.randnp.randn

넘파이의 random 모듈에는 ndarray를 랜덤한 값으로 초기화할 수 있는 함수들이 많이 있습니다. 예를 들어, 다음은 (균등 분포인) 0과 1사이의 랜덤한 부동 소수로 $3 \times 4$ 행렬을 초기화합니다:

다음은 평균이 0이고 분산이 1인 일변량 정규 분포(가우시안 분포)에서 샘플링한 랜덤한 부동 소수를 담은 $3 \times 4$ 행렬입니다:

이 분포의 모양을 알려면 맷플롯립을 사용해 그려보는 것이 좋습니다(더 자세한 것은 맷플롯립 튜토리얼을 참고하세요):

np.fromfunction

함수를 사용하여 ndarray를 초기화할 수도 있습니다:

넘파이는 먼저 크기가 (3, 2, 10)인 세 개의 ndarray(차원마다 하나씩)를 만듭니다. 각 배열은 축을 따라 좌표 값과 같은 값을 가집니다. 예를 들어, z 축에 있는 배열의 모든 원소는 z-축의 값과 같습니다:

[[[ 0.  0.  0.  0.  0.  0.  0.  0.  0.  0.]
  [ 0.  0.  0.  0.  0.  0.  0.  0.  0.  0.]]

 [[ 1.  1.  1.  1.  1.  1.  1.  1.  1.  1.]
  [ 1.  1.  1.  1.  1.  1.  1.  1.  1.  1.]]

 [[ 2.  2.  2.  2.  2.  2.  2.  2.  2.  2.]
  [ 2.  2.  2.  2.  2.  2.  2.  2.  2.  2.]]]

위의 식 x * y + z에서 x, y, z는 사실 ndarray입니다(배열의 산술 연산에 대해서는 아래에서 설명합니다). 중요한 점은 함수 my_function이 원소마다 호출되는 것이 아니고 딱 한 번 호출된다는 점입니다. 그래서 매우 효율적으로 초기화할 수 있습니다.

배열 데이터

dtype

넘파이의 ndarray는 모든 원소가 동일한 타입(보통 숫자)을 가지기 때문에 효율적입니다. dtype 속성으로 쉽게 데이터 타입을 확인할 수 있습니다:

넘파이가 데이터 타입을 결정하도록 내버려 두는 대신 dtype 매개변수를 사용해서 배열을 만들 때 명시적으로 지정할 수 있습니다:

가능한 데이터 타입은 int8, int16, int32, int64, uint8|16|32|64, float16|32|64, complex64|128가 있습니다. 전체 리스트는 온라인 문서를 참고하세요.

itemsize

itemsize 속성은 각 아이템의 크기(바이트)를 반환합니다:

data 버퍼

배열의 데이터는 1차원 바이트 버퍼로 메모리에 저장됩니다. data 속성을 사용해 참조할 수 있습니다(사용할 일은 거의 없겠지만요).

파이썬 2에서는 f.data가 버퍼이고 파이썬 3에서는 memoryview입니다.

여러 개의 ndarray가 데이터 버퍼를 공유할 수 있습니다. 하나를 수정하면 다른 것도 바뀝니다. 잠시 후에 예를 살펴 보겠습니다.

배열 크기 변경

자신을 변경

ndarrayshape 속성을 지정하면 간단히 크기를 바꿀 수 있습니다. 배열의 원소 개수는 동일하게 유지됩니다.

reshape

reshape 함수는 동일한 데이터를 가리키는 새로운 ndarray 객체를 반환합니다. 한 배열을 수정하면 다른 것도 함께 바뀝니다.

행 1, 열 2의 원소를 999로 설정합니다(인덱싱 방식은 아래를 참고하세요).

이에 상응하는 g의 원소도 수정됩니다.

ravel

마지막으로 ravel 함수는 동일한 데이터를 가리키는 새로운 1차원 ndarray를 반환합니다:

산술 연산

일반적인 산술 연산자(+, -, *, /, //, ** 등)는 모두 ndarray와 사용할 수 있습니다. 이 연산자는 원소별로 적용됩니다:

여기 곱셈은 행렬 곱셈이 아닙니다. 행렬 연산은 아래에서 설명합니다.

배열의 크기는 같아야 합니다. 그렇지 않으면 넘파이가 브로드캐스팅 규칙을 적용합니다.

브로드캐스팅

일반적으로 넘파이는 동일한 크기의 배열을 기대합니다. 그렇지 않은 상황에는 브로드캐시틍 규칙을 적용합니다:

규칙 1

배열의 랭크가 동일하지 않으면 랭크가 맞을 때까지 랭크가 작은 배열 앞에 1을 추가합니다.

여기에 (1,1,5) 크기의 3D 배열에 (5,) 크기의 1D 배열을 더해 보죠. 브로드캐스팅의 규칙 1이 적용됩니다!

규칙 2

특정 차원이 1인 배열은 그 차원에서 크기가 가장 큰 배열의 크기에 맞춰 동작합니다. 배열의 원소가 차원을 따라 반복됩니다.

(2,3) 크기의 2D ndarray(2,1) 크기의 2D 배열을 더해 보죠. 넘파이는 브로드캐스팅 규칙 2를 적용합니다:

규칙 1과 2를 합치면 다음과 같이 동작합니다:

또 매우 간단히 다음 처럼 해도 됩니다:

규칙 3

규칙 1 & 2을 적용했을 때 모든 배열의 크기가 맞아야 합니다.

브로드캐스팅 규칙은 산술 연산 뿐만 아니라 넘파이 연산에서 많이 사용됩니다. 아래에서 더 보도록 하죠. 브로드캐스팅에 관한 더 자세한 정보는 온라인 문서를 참고하세요.

업캐스팅

dtype이 다른 배열을 합칠 때 넘파이는 (실제 값에 상관없이) 모든 값을 다룰 수 있는 타입으로 업캐스팅합니다.

모든 int8uint8 값(-128에서 255까지)을 표현하기 위해 int16이 필요합니다. 이 코드에서는 uint8이면 충분하지만 업캐스팅되었습니다.

조건 연산자

조건 연산자도 원소별로 적용됩니다:

브로드캐스팅을 사용합니다:

불리언 인덱싱과 함께 사용하면 아주 유용합니다(아래에서 설명하겠습니다).

수학 함수와 통계 함수

ndarray에서 사용할 수 있는 수학 함수와 통계 함수가 많습니다.

ndarray 메서드

일부 함수는 ndarray 메서드로 제공됩니다. 예를 들면:

이 명령은 크기에 상관없이 ndarray에 있는 모든 원소의 평균을 계산합니다.

다음은 유용한 ndarray 메서드입니다:

이 함수들은 선택적으로 매개변수 axis를 사용합니다. 지정된 축을 따라 원소에 연산을 적용하는데 사용합니다. 예를 들면:

여러 축에 대해서 더할 수도 있습니다:

일반 함수

넘파이는 일반 함수(universal function) 또는 ufunc라고 부르는 원소별 함수를 제공합니다. 예를 들면 square 함수는 원본 ndarray를 복사하여 각 원소를 제곱한 새로운 ndarray 객체를 반환합니다:

다음은 유용한 단항 일반 함수들입니다:

이항 일반 함수

두 개의 ndarray에 원소별로 적용되는 이항 함수도 많습니다. 두 배열이 동일한 크기가 아니면 브로드캐스팅 규칙이 적용됩니다:

배열 인덱싱

1차원 배열

1차원 넘파이 배열은 보통의 파이썬 배열과 비슷하게 사용할 수 있습니다:

물론 원소를 수정할 수 있죠:

슬라이싱을 사용해 ndarray를 수정할 수 있습니다:

보통의 파이썬 배열과 차이점

보통의 파이썬 배열과 대조적으로 ndarray 슬라이싱에 하나의 값을 할당하면 슬라이싱 전체에 복사됩니다. 위에서 언급한 브로드캐스팅 덕택입니다.

또한 이런 식으로 ndarray 크기를 늘리거나 줄일 수 없습니다:

원소를 삭제할 수도 없습니다:

중요한 점은 ndarray의 슬라이싱은 같은 데이터 버퍼를 바라보는 뷰(view)입니다. 슬라이싱된 객체를 수정하면 실제 원본 ndarray가 수정됩니다!

데이터를 복사하려면 copy 메서드를 사용해야 합니다:

다차원 배열

다차원 배열은 비슷한 방식으로 각 축을 따라 인덱싱 또는 슬라이싱해서 사용합니다. 콤마로 구분합니다:

주의: 다음 두 표현에는 미묘한 차이가 있습니다:

첫 번째 표현식은 (12,) 크기인 1D 배열로 행이 하나입니다. 두 번째는 (1, 12) 크기인 2D 배열로 같은 행을 반환합니다.

팬시 인덱싱(Fancy indexing)

관심 대상의 인덱스 리스트를 지정할 수도 있습니다. 이를 팬시 인덱싱이라고 부릅니다.

여러 개의 인덱스 리스트를 지정하면 인덱스에 맞는 값이 포함된 1D ndarray를 반환됩니다.

고차원

고차원에서도 동일한 방식이 적용됩니다. 몇 가지 예를 살펴 보겠습니다:

어떤 축에 대한 인덱스를 지정하지 않으면 이 축의 모든 원소가 반환됩니다:

생략 부호 (...)

생략 부호(...)를 쓰면 모든 지정하지 않은 축의 원소를 포함합니다.

불리언 인덱싱

불리언 값을 가진 ndarray를 사용해 축의 인덱스를 지정할 수 있습니다.

np.ix_

여러 축에 걸쳐서는 불리언 인덱싱을 사용할 수 없고 ix_ 함수를 사용합니다:

ndarray와 같은 크기의 불리언 배열을 사용하면 해당 위치가 True인 모든 원소를 담은 1D 배열이 반환됩니다. 일반적으로 조건 연산자와 함께 사용합니다:

반복

ndarray를 반복하는 것은 일반적인 파이썬 배열을 반복한는 것과 매우 유사합니다. 다차원 배열을 반복하면 첫 번째 축에 대해서 수행됩니다.

ndarray에 있는 모든 원소를 반복하려면 flat 속성을 사용합니다:

배열 쌓기

종종 다른 배열을 쌓아야 할 때가 있습니다. 넘파이는 이를 위해 몇 개의 함수를 제공합니다. 먼저 배열 몇 개를 만들어 보죠.

vstack

vstack 함수를 사용하여 수직으로 쌓아보죠:

q1, q2, q3가 모두 같은 크기이므로 가능합니다(수직으로 쌓기 때문에 수직 축은 크기가 달라도 됩니다).

hstack

hstack을 사용해 수평으로도 쌓을 수 있습니다:

q1과 q3가 모두 3개의 행을 가지고 있기 때문에 가능합니다. q2는 4개의 행을 가지고 있기 때문에 q1, q3와 수평으로 쌓을 수 없습니다:

concatenate

concatenate 함수는 지정한 축으로도 배열을 쌓습니다.

예상했겠지만 hstackaxis=1으로 concatenate를 호출하는 것과 같습니다.

stack

stack 함수는 새로운 축을 따라 배열을 쌓습니다. 모든 배열은 같은 크기를 가져야 합니다.

배열 분할

분할은 쌓기의 반대입니다. 예를 들어 vsplit 함수는 행렬을 수직으로 분할합니다.

먼저 6x4 행렬을 만들어 보죠:

수직으로 동일한 크기로 나누어 보겠습니다:

split 함수는 주어진 축을 따라 배열을 분할합니다. vsplitaxis=0으로 split를 호출하는 것과 같습니다. hsplit 함수는 axis=1split를 호출하는 것과 같습니다:

배열 전치

transpose 메서드는 주어진 순서대로 축을 뒤바꾸어 ndarray 데이터에 대한 새로운 뷰를 만듭니다.

예를 위해 3D 배열을 만들어 보죠:

0, 1, 2(깊이, 높이, 너비) 축을 1, 2, 0 (깊이→너비, 높이→깊이, 너비→높이) 순서로 바꾼 ndarray를 만들어 보겠습니다:

transpose 기본값은 차원의 순서를 역전시킵니다:

넘파이는 두 축을 바꾸는 swapaxes 함수를 제공합니다. 예를 들어 깊이와 높이를 뒤바꾸어 t의 새로운 뷰를 만들어 보죠:

선형 대수학

넘파이 2D 배열을 사용하면 파이썬에서 행렬을 효율적으로 표현할 수 있습니다. 주요 행렬 연산을 간단히 둘러 보겠습니다. 선형 대수학, 벡터와 행렬에 관한 자세한 내용은 Linear Algebra tutorial를 참고하세요.

행렬 전치

T 속성은 랭크가 2보다 크거나 같을 때 transpose()를 호출하는 것과 같습니다:

T 속성은 랭크가 0이거나 1인 배열에는 아무런 영향을 미치지 않습니다:

먼저 1D 배열을 하나의 행이 있는 행렬(2D)로 바꾼다음 전치를 수행할 수 있습니다:

행렬 곱셈

두 개의 행렬을 만들어 dot 메서드로 행렬 곱셈을 실행해 보죠.

주의: 앞서 언급한 것처럼 n1*n2는 행렬 곱셈이 아니라 원소별 곱셈(또는 아다마르 곱이라 부릅니다)입니다.

역행렬과 유사 역행렬

numpy.linalg 모듈 안에 많은 선형 대수 함수들이 있습니다. 특히 inv 함수는 정방 행렬의 역행렬을 계산합니다:

pinv 함수를 사용하여 유사 역행렬을 계산할 수도 있습니다:

단위 행렬

행렬과 그 행렬의 역행렬을 곱하면 단위 행렬이 됩니다(작은 소숫점 오차가 있습니다):

eye 함수는 NxN 크기의 단위 행렬을 만듭니다:

QR 분해

qr 함수는 행렬을 QR 분해합니다:

행렬식

det 함수는 행렬식을 계산합니다:

고윳값과 고유벡터

eig 함수는 정방 행렬의 고윳값과 고유벡터를 계산합니다:

특잇값 분해

svd 함수는 행렬을 입력으로 받아 그 행렬의 특잇값 분해를 반환합니다:

svd 함수는 Σ의 대각 원소 값만 반환합니다. 전체 Σ 행렬은 다음과 같이 만듭니다:

대각원소와 대각합

선형 방정식 풀기

solve 함수는 다음과 같은 선형 방정식을 풉니다:

solution을 확인해 보죠:

좋습니다! 다른 방식으로도 solution을 확인해 보죠:

벡터화

한 번에 하나씩 개별 배열 원소에 대해 연산을 실행하는 대신 배열 연산을 사용하면 훨씬 효율적인 코드를 만들 수 있습니다. 이를 벡터화라고 합니다. 이를 사용하여 넘파이의 최적화된 성능을 활용할 수 있습니다.

예를 들어, $sin(xy/40.5)$ 식을 기반으로 768x1024 크기 배열을 생성하려고 합니다. 중첩 반복문 안에 파이썬의 math 함수를 사용하는 것은 나쁜 방법입니다:

작동은 하지만 순수한 파이썬 코드로 반복문이 진행되기 때문에 아주 비효율적입니다. 이 알고리즘을 벡터화해 보죠. 먼저 넘파이 meshgrid 함수로 좌표 벡터를 사용해 행렬을 만듭니다.

여기서 볼 수 있듯이 XY 모두 768x1024 배열입니다. X에 있는 모든 값은 수평 좌표에 해당합니다. Y에 있는 모든 값은 수직 좌표에 해당합니다.

이제 간단히 배열 연산을 사용해 계산할 수 있습니다:

맷플롯립의 imshow 함수를 사용해 이 데이터를 그려보죠(matplotlib tutorial을 참조하세요).

저장과 로딩

넘파이는 ndarray를 바이너리 또는 텍스트 포맷으로 손쉽게 저장하고 로드할 수 있습니다.

바이너리 .npy 포맷

랜덤 배열을 만들고 저장해 보죠.

끝입니다! 파일 이름의 확장자를 지정하지 않았기 때문에 넘파이는 자동으로 .npy를 붙입니다. 파일 내용을 확인해 보겠습니다:

이 파일을 넘파이 배열로 로드하려면 load 함수를 사용합니다:

텍스트 포맷

배열을 텍스트 포맷으로 저장해 보죠:

파일 내용을 확인해 보겠습니다:

이 파일은 탭으로 구분된 CSV 파일입니다. 다른 구분자를 지정할 수도 있습니다:

이 파일을 로드하려면 loadtxt 함수를 사용합니다:

압축된 .npz 포맷

여러 개의 배열을 압축된 한 파일로 저장하는 것도 가능합니다:

파일 내용을 확인해 보죠. .npz 파일 확장자가 자동으로 추가되었습니다.

다음과 같이 이 파일을 로드할 수 있습니다:

게으른 로딩을 수행하는 딕셔너리와 유사한 객체입니다:

그 다음은?

넘파이 기본 요소를 모두 배웠지만 훨씬 더 많은 기능이 있습니다. 이를 배우는 가장 좋은 방법은 넘파이를 직접 실습해 보고 훌륭한 넘파이 문서에서 필요한 함수와 기능을 찾아 보세요.