NumPy로 행렬의 행과 열 개수 구하는 방법

이 글에서는 NumPy를 사용해 행렬을 생성한 뒤, 해당 행렬의 행(row)과 열(column)의 개수를 확인하는 방법을 알아보겠습니다.

알고리즘

1단계: 난수로 이루어진 NumPy 행렬을 생성합니다.
2단계: numpy.shape 함수를 사용하여 행렬의 행과 열 정보를 가져옵니다.
3단계: 행의 개수와 열의 개수를 출력합니다.

예제 코드

아래 예제에서는 np.random.rand() 함수를 사용해 2행 3열 크기의 무작위 실수 행렬을 만들고, shape 속성으로 크기를 확인합니다.

import numpy as np

matrix = np.random.rand(2, 3)
print(matrix)
print("주어진 행렬의 행과 열 개수는: ", matrix.shape)

실행 결과

[[0.23226052 0.89690884 0.19813164]
 [0.85170808 0.97725669 0.72454096]]
주어진 행렬의 행과 열 개수는:  (2, 3)

참고 사항

matrix.shape는 튜플(tuple) 형태로 결과를 반환하며, 첫 번째 값은 행의 개수, 두 번째 값은 열의 개수를 나타냅니다. 따라서 위 예제에서 반환된 값 (2, 3)은 행렬이 2개의 행과 3개의 열로 구성되어 있음을 의미합니다.

또한 행 또는 열의 개수만 개별적으로 얻고 싶다면 다음과 같이 인덱싱을 활용할 수 있습니다.

rows = matrix.shape[0]   # 행의 개수
cols = matrix.shape[1]   # 열의 개수