넘파이, 파이썬 어레이 0보다 큰 값 찾기(numpy array masking)

넘파이, 파이썬 어레이 0보다 큰 값 찾기(numpy array masking)

 

파이썬 어래이에서 특정 조건에 맞는 값을 추출 하기위해서 마스킹 기법을 사용합니다.

마스킹 사용 예시 코드는 아래와 같습니다.

 

import numpy as np # 예시 어레이 생성 float_example = np.array( [1.45, -3.1, 3.653, 6.345, -1.23] ) # 0보다 크면 True 작으면 False를 반환하는 마스킹 생성 mask = float_example > 0 # 마스킹을 이용하여 0이상을 숫자만 인덱싱 float_example = float_example[mask] print(float_example)

 

위 코드를 실행하면 아래 그림과 같은 결과를 도출 할 수 있습니다.

 

일반적으로, 간결히 사용하기 위해 mask 변수를 따로 도출하지 않고 빨간 글씨처럼 코드를 구성합니다.

import numpy as np # 예시 어레이 생성 float_example = np.array( [1.45, -3.1, 3.653, 6.345, -1.23] ) float_example = float_example[float_example > 0] print(float_example)