[Python] 파이썬 Numpy(넘파이) 배열 마스킹

마스킹이란

원하는 조건을 달아서 조건에 맞는 배열요소들만 걸러내는 작업이다.

 

True/False를 반환하는 조건을 인덱스처럼 집어 넣으면 조건의 True 값을 반환하는 원소들만 뽑아낼 수 있다.

 

arrra2 = array[array > 3]

 

import numpy as np
my_arr = np.random.randint(0,10,(3,4))

[[6 3 5 6]
 [2 3 4 3]
 [0 4 8 0]]

print(my_arr > 5)
# 결과 
[[ True False False  True]
 [False False False False]
 [False False  True False]]

mask_arr = my_arr[my_arr >5] 
# 결과
[6 6 8]