Come filtrare una matrice numpy usando i valori di un'altra matrice?

Question

Ho due array NumPy, per esempio:

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

e una matrice di filtro, ad esempio:

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

len(a) == len(f) 

Come posso ottenere un nuovo array NumPy con solo i valori in cui lo stesso indice f è vero? Nel mio caso: [2, 5].

Secondo la soluzione accettata (con valori diversi):

>>> 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 supporta boolean indexing

a[f] 

Questo presuppone che a e f sono array NumPy piuttosto che liste Python (come nella questione). È possibile convertire con f = np.array(f).

Answer 2

Se non si dispone già bisogno di array NumPy, ecco un elenco semplice:

import itertools 
print itertools.compress(a, f) 

Per pre-2.7 versioni di Python, è necessario roll your own (vedi manuale):

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