我有一个numpy。ndarray和我有一个布尔列表。我想使用列表访问数组中的列。
X = [[1,2,3,4],[5,6,7,8]] Y = [True,False,False,True]
我希望结果是
[[1,4][5,8]]
我想我这样做效率很低,我想知道是否有一个简单的方法。
你必须先把它转换成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])