Python numpy : ndim (array의 차원(dimention) 출력하기)

 

 

 

numpy의 ndim attribute는 어떤 array가 몇차원인지를 알려줍니다.

 

아래는 ndim attribute를 사용한 예시입니다.

 

import numpy as np


arr_test = np.array(
    [1, 2, 3, 4, 5]  # 1차 array
)
print(arr_test)
print(arr_test.ndim)



-- Result
[1 2 3 4 5]
1

 

 

import numpy as np


arr_test = np.array(
    [  # 1차 array
        [1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5]  # 2차 array
    ]
)
print(arr_test)
print(arr_test.ndim)



-- Result
[[1 2 3 4 5]
 [1 2 3 4 5]
 [1 2 3 4 5]]
2

 

 

 

import numpy as np


arr_test = np.array(
    [  # 1차 array
        [  # 2차 array
            [1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5]  # 3차 array
        ],
        [  # 2차 array
            [1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5]  # 3차 array
        ],
    ]
)
print(arr_test)
print(arr_test.ndim)



-- Result
[[[1 2 3 4 5]
  [1 2 3 4 5]
  [1 2 3 4 5]]

 [[1 2 3 4 5]
  [1 2 3 4 5]
  [1 2 3 4 5]]]
3