我有两个NumPy数组,例如:
a = [1,2,3,4,5]
Run Code Online (Sandbox Code Playgroud)
和过滤器数组,例如:
f = [False, True, False, False, True]
len(a) == len(f)
Run Code Online (Sandbox Code Playgroud)
如何才能获得一个新的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])
Run Code Online (Sandbox Code Playgroud)