더북(TheBook)

12. numpy 사용해보기

# numpy의 array 함수를 쓰기 위해 numpy 패키지를 np라는 이름으로 불러온다
>>> import numpy as np
>>> a = np.array([0, 1, 2, 3, 4, 5])
>>> a.ndim # ndim은 a의 차원 수를 알려준다
1
# 배열 a의 각 차원에 몇 개의 값이 있는지를 튜플로 반환한다. 값 6개를 갖는 1차원 배열은 다음처럼 표현된다
>>> a.shape
(6,)

# 배열 a의 값들의 타입을 알려주며, 아래 결과는 정수(integer)임을 의미한다
>>> a.dtype
dtype('int32')
# 기본 인덱싱
>>> a[[2, 3, 4]]
array([2, 3, 4])
# 배열을 인덱싱으로 사용할 수 있다
>>> a[np.array([2, 3, 4])]
array([2, 3, 4])

# 배열 중 원하는 값 출력을 위해 논리 연산하기
>>> a > 4
array([False, False, False, False, False, True])
>>> a[a > 4]
array([5])
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.