A.1.1 배열과 행렬 만들기
배열과 행렬을 만드는 여러 가지 방법이 있지만, 그중 대표적인 몇 가지를 간단히 살펴보면 다음과 같다.
리스트를 먼저 만들고 array 함수를 사용해 배열을 만든다.
vec = [ 1, 2, 3, 4 ] # 네 개의 원소를 가진 리스트 vec를 만든다
vecA = np.array( vec ) # 리스트를 배열로 바꾼다
print( vecA )
결과
[1 2 3 4]
array 함수로 행렬도 만들 수 있다. [ [ ],[ ] ] 형태로 다소 복잡해 보이지만, 안쪽의 []가 행을 의미한다. [1,2,3]과 [4,5,6]은 행이다. 그리고 열은 콤마로 구분된다.
vecB = np.array( [ [1, 2, 3], [4, 5, 6] ] ) # array( ) 함수에 직접 리스트를 입력한다
print( vecB )
결과
[[1 2 3] [4 5 6]]