考虑以下代码
a = numpy.array([[1,2,3],[4,5,6],[7,8,9]])
a[[True,True,False],[True,True,False]]
我期望的是一个子矩阵
array([[1, 2],
[4, 5]])
但我得到的是
array([1, 5])
我找到了一种检索子矩阵的方法
a[np.ix_([True,True,False],[True,True,False])]
我的问题:
我们能以更优雅的方式做到这一点吗?