I have two NumPy arrays, e.g.:
a = [1,2,3,4,5]
and a filter array, e.g.:
f = [False, True, False, False, True]
len(a) == len(f)
How can I get a new numpy array with only the values in a where the same index in f
is True? In my case: [2, 5]
.
According to the accepted solution (with different values):
>>> 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])
See Question&Answers more detail:
os 与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…