[python] NumPy 배열의 슬라이싱

14 Dec 2023

NumPy는 파이썬에서 과학적 계산을 수행하기 위한 핵심 라이브러리 중 하나입니다. NumPy 배열은 다차원 배열을 다루는데 효율적이며, 배열의 일부분을 추출하는 기능인 슬라이싱은 매우 유용합니다.

1. 1차원 배열의 슬라이싱

1차원 NumPy 배열에서의 슬라이싱은 파이썬의 리스트 슬라이싱과 유사합니다. 배열의 특정 부분을 추출하여 새로운 배열을 생성할 수 있습니다.

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
sliced_arr = arr[1:4]

print(sliced_arr)

위 예제에서 arr[1:4]는 인덱스 1부터 3까지 (4는 포함되지 않음)의 요소를 포함하는 새로운 배열을 생성합니다.

다차원 NumPy 배열에서의 슬라이싱은 각 차원별로 슬라이스 범위를 지정하여 배열의 일부분을 추출할 수 있습니다.

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 두 번째와 세 번째 행을 선택하고, 두 번째 열 이후의 요소들을 선택
sliced_arr = arr[1:, 1:]

print(sliced_arr)

위 예제에서 arr[1:, 1:]은 인덱스 1부터 끝까지의 행을 선택하고, 인덱스 1부터 끝까지의 열을 선택하여 새로운 배열을 생성합니다.

NumPy 배열의 슬라이싱을 통해 데이터를 효율적으로 처리하고 부분집합을 추출할 수 있습니다.

참고문헌: