Numpy - 배열 생성

이수안컴퓨터연구소 - Numpy 한 번에 끝내기 영상을 본 뒤 복습용으로 기록하는 공간 :)

 

numpy 버전 확인

import numpy as np
np.__version__

'1.20.3'

리스트로 배열 만들기

a1=np.array([1,2,3,4,5])
a1
print(type(a1))
print(a1.shape)
print(a1[0], a1[1], a1[2], a1[3], a1[4])
a1[0]=4
a1[1]=5
a1[2]=6 #이런 식으로 수정도 가능
a1

<class 'numpy.ndarray'>
(5,)
1 2 3 4 5

 

a2=np.array([[1,2,3],[4,5,6],[7,8,9]]) #2차원 배열
a2
print(type(a2))
print(a2.shape)
print(a2[0,0],a2[1,1],a2[2,2],a2[1,2])

<class 'numpy.ndarray'>
(3, 3)
1 5 9 6

 

a3=np.array([[[1,2,3],[4,5,6],[7,8,9]],
             [[1,2,3],[4,5,6],[7,8,9]],
             [[1,2,3],[4,5,6],[7,8,9]]
            ])
a3
print(type(a3))
print(a3.shape)
print(a3[0,0,0], a3[1,1,1], a3[2,2,2],[1,2,])

<class 'numpy.ndarray'>
(3, 3, 3)
1 5 9 [1, 2]

배열 생성 및 초기화

 

배열을 0으로 초기화

np.zeros(10)

array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0.])

 

1로 초기화 된 3x3 2차원 배열

np.ones((3,3))

array([[1., 1., 1.],
       [1., 1., 1.],
       [1., 1., 1.]])

 

full(): 모든 요소를 지정한 값으로 초기화

np.full((3,3), 1.23)

array([[1.23, 1.23, 1.23],
       [1.23, 1.23, 1.23],
       [1.23, 1.23, 1.23]])