내가 NumPy와 배열 벡터가 인덱스

Question

에 따라 NumPy와 배열에서 부분 배열을 가져, 나는 인덱스에 따라 일부 싶어 : [= 출력을 얻을 수있는 파이썬 방법이 무엇인지

import numpy as np 
input=np.array([1,2,3,4,5,6,7,8,9,10]) 
index=np.array([0,1,0,0,0,0,1,0,0,1]) 

을 2,7,10]?

Answer 1

output = input[index.astype(np.bool)] 

또는

output = input[np.where(index)[0]]