NumPy 배열의 차원 확인하기

NumPy 배열이 1차원인지 2차원인지 확인하는 방법은 배열의 shape 속성을 확인하면 된다. shape 속성은 배열의 차원을 나타내는 튜플(tuple)을 반환한다.

 

- 1차원 배열인 경우:
shape 값은 (n,) 형태로 표현된다. n은 배열의 요소 개수이다.

 

- 2차원 배열인 경우:
shape 값은 (m, n) 형태로 표현된다. m은 행(row)의 개수, n은 열(column)의 개수이다.

 

import numpy as np

# 예시 1: 1차원 배열
arr_1d = np.array([1, 2, 3, 4, 5])
print(arr_1d.shape)  # 출력: (5,) -> 1차원 배열

# 예시 2: 2차원 배열
arr_2d = np.array([[1, 2], [3, 4], [5, 6]])
print(arr_2d.shape)  # 출력: (3, 2) -> 2차원 배열