Numpy - Shape Handling (reshape, flatten)

>

reshape

shape을 바꾸기 위해 사용

import numpy as np

In

test_matrix = [[1,2,3,4], [1,2,5,8]]
np.array(test_matrix).shape

shape 정의

Out

(2, 4)

예시
In

np.array(test_matrix).reshape(2,2,2)

Out

array([[[1, 2],
        [3, 4]],

       [[1, 2],
        [5, 8]]])

주로 쓰이는 것
In

np.array(test_matrix).reshape(-1,2).shape

Out

(4, 2)

row -1 : size를 기반으로 row 개수 선정하겠다는 의미
만약 test_matrix의 총 데이터 개수가 100개라면
size가 2가 되게하도록 하는 row = 50
즉
(50,2)가 나온다.

>

flatten

다차원 array를 1차원 array로 변경
In

test_matrix = [[[1,2,3,4], [1,2,5,8]], [[1,2,3,4], [1,2,5,8]]]
np.array(test_matrix).flatten()

Out

array([1, 2, 3, 4, 1, 2, 5, 8, 1, 2, 3, 4, 1, 2, 5, 8])