행렬 를 만들고 몇 가지 행렬 연산을 수행해보자.
import numpy as np
# 대괄호를 이용해 행렬을 만든다
a = np.matrix( [ [1, 2], [3, 4] ] )
# 작은따옴표를 사용해 다음과 같이 만들 수도 있다
a = np.matrix( '1 2; 3 4' )
# +, -, *, / 연산자를 사용해 행렬 덧셈, 뺄셈, 곱셈, 나눗셈 연산을 한다
ap = a + a # 행렬 간 원소끼리 더한다
am = a - a # 행렬 간 원소끼리 뺀다
aa = a * a # 행렬 곱셈을 한다
ad = a / a # 행렬 간 원소끼리 나눈다
# matrix.T 함수를 사용해 전치행렬을 구한다
aT = a.T
# matrix.I 함수를 사용해 역행렬을 구한다
aI = a.I
# 콜론(:)은 행렬의 행이나 열 전체를 의미한다
# 행이나 열의 인덱스는 0부터 시작한다
# 행렬[ 행 , 열 ] 같은 방식으로 행렬의 일부를 지정할 수 있다
row0 = a[ 0 , : ] # 첫 번째 행의 모든 열의 값, 즉 첫 번째 행 전체를 구한다
row1 = a[ 1 , : ] # 두 번째 행의 모든 열의 값, 즉 두 번째 행 전체를 구한다
col0 = a[ : , 0 ] # 첫 번째 열의 모든 행의 값, 즉 첫 번째 열 전체를 구한다
col1 = a[ : , 1 ] # 두 번째 열의 모든 행의 값, 즉 두 번째 열 전체를 구한다