[파이썬 numpy] 배열 전치하는 법 (T 메소드)

[파이썬 numpy] 

배열 전치하는 법 (T 메소드)

배열을 하나 정의합시다. 

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

2행 3열인 배열입니다. 

>>> A

array([[1, 2, 3],

       [4, 5, 6]])

위 배열의 전치배열을 만들 때는 T 메소드를 사용합니다. 

>>> A.T

array([[1, 4],

       [2, 5],

       [3, 6]])

고차원 배열에도 적용해봅시다. 

A2=np.arange(24).reshape(2,3,4)

3행4열짜리 배열이 2층 쌓여있다고 해석할 수 있습니다. 

>>> A2

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

        [ 4,  5,  6,  7],

        [ 8,  9, 10, 11]],

       [[12, 13, 14, 15],

        [16, 17, 18, 19],

        [20, 21, 22, 23]]])

전치를 적용해봅시다. 

>>> A2.T

array([[[ 0, 12],

        [ 4, 16],

        [ 8, 20]],

       [[ 1, 13],

        [ 5, 17],

        [ 9, 21]],

       [[ 2, 14],

        [ 6, 18],

        [10, 22]],

       [[ 3, 15],

        [ 7, 19],

        [11, 23]]])

3행 2열짜리 배열이 4층 쌓인 것으로 바뀌었습니다. shape을 비교해보면 2,3,4 에서 4,3,2로 바뀐것을 알 수 있습니다. 

>>> A2.T.shape

(4, 3, 2)