본문 바로가기

BE/Python

[Python] Numpy 배열 함수

Numpy 배열 생성하기  

Numpy란? 다차원 배열을 처리하는 데 필요한 여러 기능을 정의해놓은 파이썬 라이브러리입니다. 

Numpy는 배열에 동일한 타입의 값(데이터)만 삽입해야 합니다. 

Numpy를 사용하기 위해 라이브러리를 import합니다. 

import numpy as np

 

Numpy 라이브러리 함수 - 배열 생성 

1. array() : 동일한 타입의 데이터 배열 생성하기

import numpy as np
array = np.array([0,1,2,3,4,5])

 

 

2. zeros() : 0이 들어있는 배열 생성하기 

import numpy as np
array = np.zeros(2)
print(array)

 

결과) 배열의 모든 값이 0으로 초기화된 것을 볼 수 있습니다.

[0 0]

 

 

3. ones() : 1이 들어있는 배열 생성하기 

import numpy as np
array = np.ones(2)
print(array)

 

결과) 배열의 모든 값이 1로 초기화된 것을 볼 수 있습니다.

[1 1]

 

4. arrange(n) : N이하의 값으로 초기화(배열 버전의 range 함수) 

import numpy as np
array = np.arrange(5)
print(array)

 

결과) 크기 5의 배열이 0부터 4까지 초기화된 것을 볼 수 있습니다.

[0 1 2 3 4]

 

5. random.randint(start, end, (row* col )) : 설정한 범위 안에서 랜덤하게 수를 생성하는 배열 만들기  

예시)  0이상 5미만의 랜덤한 값이 들어가 있는 2*3 배열을 만듭니다. 

array = np.random.randint(0,5,(2,3))

 

결과) 0이상 5미만의 수가 배열에 저장된 것을 볼 수 있습니다 .

[[1 0 3]
[2 3 4]]

 

Numpy 라이브러리 함수 - 배열 기초 

1. array.size : 배열 크기 출력

예시) 배열의 크기(10)를 출력합니다. 

array = np.arange(10)
print(array.size)
10

 

2. array[start: end+1] : start부터 end까지 요소 출력

예시) 2부터 4까지 요소를 출력하고 싶은 경우

array = np.arange(10) 
print(array[2,5])

결과 ) 2부터 4까지 출력됩니다.

[2 3 4]

 

3. array.ndim : 배열 차원 출력 

array = np.arange(10)
print(array.ndim)

 

결과) 배열이 1차원이므로 1을 출력합니다.

1

 

4. np.arange(start, end).reshape : 2차원 배열 만들기 

arange: range와 동일하게 연속된 숫자들을 만듭니다. 

reshape : array의 shape을 변경합니다.

matrix = np.arange(1, 30).reshape(2,4)
#1부터 20까지의 수가 들어있는 2*4배열 

 

5. np.concatenate([arr1, arr2]) : 배열 합치기 

두 배열을 axis 방향으로 이어붙입니다.  axis = 0 : 세로 방향 , axis =1 : 가로 방향

matrix = np.arange(1, 30).reshape(2,2)
np.concatenate([matrix, matrix], axis = 0)