使用布尔向量索引numpy数组

Question

我有一个numpy.ndarray，我有一个布尔列表。我想使用列表来访问数组中的列。

X = [[1,2,3,4],[5,6,7,8]]
Y = [True,False,False,True]

我希望结果是

[[1,4][5,8]]

我想我这样做效率不高，想知道是否有一种简单明了的方法。

Answer 1

您必须先将其转换为numpy。

import numpy as np

X = np.array([[1,2,3,4],[5,6,7,8]])
Y = np.array([True,False,False,True])

print(X[:,Y])