다른 배열의 값을 사용하여 numpy 배열을 필터링하는 방법은 무엇입니까?

Question

나는 예를 들어 두 NumPy와 배열을 가지고 :

a = [1,2,3,4,5] 

및 필터 배열을, 예를 들면 : 나는에서만 값으로 새로운 NumPy와 배열을 얻을 수있는 방법

f = [False, True, False, False, True] 

len(a) == len(f) 

어디 같은 지수 f이 참입니까? 내 경우에는 : [2, 5]. (다른 값) 허용 솔루션에 따르면

:

>>> a = numpy.array([ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) 
>>> b = numpy.array([True, False, True, False, True, False, True, False, True, False]) 
>>> a[b] 
array([1, 3, 5, 7, 9]) 

Answer 1

NumPy와는이 a 및 f이 문제로 (NumPy와 배열이 아닌 파이썬 목록 있다고 가정 boolean indexing

a[f] 

지원). f = np.array(f)으로 변환 할 수 있습니다. 이미 NumPy와 배열이 필요하지 않은 경우

Answer 2

, 여기에 일반 목록입니다 : 파이썬의 사전 2.7 버전의

import itertools 
print itertools.compress(a, f) 

, 당신은 당신의 자신의 (매뉴얼 참조) 롤해야합니다

def compress(data, selectors): 
    return (d for d, s in itertools.izip(data, selectors) if s) 

을