通过布尔掩码数组选择 numpy 数组的元素

我有一个布尔掩码数组a长度n:

a = np.array([True, True, True, False, False])

我有一个二维数组n列：

b = np.array([[1,2,3,4,5], [1,2,3,4,5]])

例如，我想要一个仅包含“True”值的新数组

c = ([[1,2,3], [1,2,3]])

c = a * b不起作用，因为它还包含我不想要的假列“0”

c = np.delete(b, a, 1) does not work

有什么建议么？

你可能想要这样的东西：

>>> a = np.array([True, True, True, False, False])
>>> b = np.array([[1,2,3,4,5], [1,2,3,4,5]])
>>> b[:,a]
array([[1, 2, 3],
       [1, 2, 3]])

请注意，要使这种索引起作用，它需要是ndarray，就像你正在使用的那样，而不是list，或者它会解释False and True as 0 and 1并给你这些列：

>>> b[:,[True, True, True, False, False]]   
array([[2, 2, 2, 1, 1],
       [2, 2, 2, 1, 1]])