Numpy 기본 2

Numpy 행렬 형태 바꾸기

Numpy의 여러 함수를 사용하여 원하는 형태의 행렬을 만들 수 있습니다.

.concatenate()

# numpy이를 사용하기 위해 불러온다.
# numpy를 np라고 사용할 수 있다.
import numpy as np

array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])

# concatenate()를 이용해서 array1과 array2를 붙여준다.
array3 = np.concatenate([array1, array2])

print(array3)

결과
[1 2 3 4 5 6]

.reshape()

array1 = np.array([1, 2, 3, 4])

# 2x2 형태로 바꿔서 저장한다.
array2 = array1.reshape((2, 2))
print(array1)
print(array2)

결과
[1 2 3 4]

[[1 2][3 4]] ( 2x2 행렬)

.split()

array = np.arange(8).reshape(2, 4)

# split을 사용해서 행렬을 나눌 수 있다.
# 분리한 후 넣어 줄 변수, 나눌 행렬, 나누는 index 기준, axis를 넣어주면 된다.
left, right = np.split(array, [2], axis=1)

print(array)
print(left)
print(right)

결과
[[0 1 2 3][4 5 6 7]]

[[0 1][4 5]]

[[2 3][6 7]]