더북(TheBook)

그리고 마스크를 다음과 같이 변수에 넣어서 사용할 수도 있습니다.

mask1 = abs(a) > 3
print(a[mask1])

 

실행 결과

[-5 –4  4]

 

TIP

여기에서 ‘abs(a) > 3’은 ‘a 배열에 저장된 원소의 절대값이 3보다 크다’라는 의미입니다.

또 이렇게 몇 개의 마스크를 연결해서 사용할 수도 있습니다.

mask1 = abs(a) > 3
mask2 = abs(a) % 2 == 0
print(a[mask1+mask2])   # 둘 중 하나의 조건이라도 참일 경우
print(a[mask1*mask2])   # 두 가지 조건이 모두 참일 경우

 

실행 결과

[-5 -4 -2  0  2  4]
[-4  4]

이 외에도 마스크는 다양하게 활용될 수 있는데요. 그림 13-7에서 만들었던 버블 차트에 마스크를 적용하면 어떤 그래프가 그려질지 직접 확인하세요!

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.