numpy 조건에 맞는 값들만 뽑아내기

b > 3 를 만족하는 인덱스를 a 에서 뽑아내기

a = np.array([1,1,1,2,2,2]) b = np.array([4,6,1,8,2,1])

 

• 방법-1

new_a = a[ b > 3 ]

 

• 방법-2

new_a = a[ np.where(b>3)[0] ]