SeouliteLab

[Python/파이썬] Numpy shape() 함수, 배열 크기/형태/차원 확인하기 본문

프로그래밍

[Python/파이썬] Numpy shape() 함수, 배열 크기/형태/차원 확인하기

Seoulite Lab 2024. 3. 1. 16:30

Numpy의 `shape()` 함수는 주어진 배열의 크기, 형태, 차원을 확인하는 데 사용됩니다. 이 함수를 사용하여 배열의 모양을 확인하는 방법에 대해 알아보겠습니다.

예제 1: 1차원 배열의 크기 확인

가장 간단한 경우는 1차원 배열의 크기를 확인하는 것입니다. 아래 예제에서는 이를 확인할 수 있습니다.

import numpy as np

# 1차원 Numpy 배열 생성
my_array = np.array([1, 2, 3, 4, 5])

# 배열의 크기 확인
result = my_array.shape

print(result)

위의 코드에서는 `shape()` 함수를 사용하여 1차원 Numpy 배열 `my_array`의 크기를 확인합니다.

예제 2: 다차원 배열의 크기 확인

다차원 배열의 경우 배열의 모양이 더 복잡해집니다. 아래 예제에서는 다차원 배열에 대한 크기 확인 방법을 살펴봅니다.

import numpy as np

# 다차원 Numpy 배열 생성
my_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 배열의 크기 확인
result = my_array.shape

print(result)

위의 코드에서는 다차원 Numpy 배열의 크기를 확인합니다. `shape()` 함수는 배열의 행과 열의 수를 반환합니다.