使用掩码在numpy中检索子矩阵

时间:2018-01-26 01:31:54

标签: python numpy

考虑以下代码

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])]

我的问题:

  

我们能以更优雅的方式做到这一点吗?

0 个答案:

没有答案