如何使用另一个数组的值过滤numpy数组？

Question

我有两个NumPy数组,例如:

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

和过滤器数组,例如:

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

len(a) == len(f)

如何才能获得一个新的numpy数组,其中只有相同索引的f值为True？就我而言:[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]

这假设是 a和fNumPy数组而不是Python列表(如问题中所示).你可以转换f = np.array(f).