[Python] numpy 행렬 계산

numpy 행렬 표현

import numpy as np

A = np.array([[0, 1, -1], 
            [1, 2, 3], 
            [2, 1, 0], 
            [-1, 2, -4]
            ])

B = np.array([
            [0, 2], 
            [1, 1],
            [-1, -2]
            ])

print(A)
print(B)
print(A[1][1])  # A의 2행 2열
print(A[3][0])  # A의 4행 1열

 

numpy 행렬 덧셈

import numpy as np

a = np.array([
                [1, 2, -1], 
                [1, 1, -3]
            ])
            
b = np.array([
                [3, 2, -2], 
                [-1, 2, 1]
            ])
            
print(a + b)

#[[ 4  4 -3]
# [ 0  3 -2]]

 

numpy 행렬 곱셈

- 스칼라 곱

아래의 예제는 i 의 값이 5일 때 a 행렬과 곱하는 예제이다.

import numpy as np

a = np.array([
                [1, 2, -1], 
                [1, 1, -3]
            ])
            
print(5 * a)

 

- 행렬 X 행렬

행렬에 곱에는 내적과 외적의 곱이있다.

내적은 np.dot, a @ b로 하여 구할 수 있으며, 외적의 곱은 나중에 살펴보자

# 방법 1
np.dot(a, b)

# 방법 2
a @ b

 

  - 요소별 곱하기

import numpy as np

A = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

B = np.array([
    [0, 1, 2],
    [2, 0, 1],
    [1, 2, 0]
])

A * B