본문 바로가기
Python/numpy

파이썬 numpy.extract 함수 활용하기

by PySun 2024. 6. 25.
반응형

Numpy Extract 함수 소개:

Numpy의 `numpy.extract` 함수는 주어진 조건을 만족하는 요소를 추출하는 함수입니다. 이 함수는 조건을 만족하는 요소를 새로운 배열로 반환합니다. 데이터 필터링 및 특정 조건을 만족하는 요소를 선택하는 데에 사용됩니다.

기본 사용법:

`numpy.extract` 함수의 기본 사용법은 다음과 같습니다.

import numpy as np

# 조건에 따라 요소 추출하는 예제
array = np.array([1, 2, 3, 4, 5])
condition = array > 3
result = np.extract(condition, array)
print("조건을 만족하는 요소 추출 결과:", result)

위 예시 코드에서 `np.extract(condition, array)`는 주어진 조건을 만족하는 요소를 추출하여 반환합니다.

예시 코드:

  • 다차원 배열에서 조건에 따라 요소 추출하는 예제:
# 다차원 배열에서 조건에 따라 요소 추출하는 예제
array_2d = np.array([[1, 2, 3], [4, 5, 6]])
condition_2d = array_2d > 3
result_2d = np.extract(condition_2d, array_2d)
print("2차원 배열에서 조건을 만족하는 요소 추출 결과:", result_2d)

결론:

`numpy.extract` 함수는 조건에 따라 요소를 추출하여 새로운 배열로 반환하는 유용한 함수입니다. 이를 통해 데이터 필터링 및 특정 조건을 만족하는 요소를 선택하는 데에 활용할 수 있습니다. 다양한 예시 코드를 통해 `numpy.extract` 함수의 활용법을 익혀보세요.

반응형