我是Python的新手,但我有使用Matlab的经验。
在我的代码中,我想使用两个1D数组下标2D数组,如下所示:
a=np.array([[1,2,3],[4,5,6],[7,8,9]])
patch=np.arange(2)
b = a [patch]
c= a[patch] [patch]
因此,我希望有一个2x2矩阵:[(1, 2) (4, 5)]
。相反,我的b
和c
是2x3矩阵:b=c=[(1, 2, 3), (4,5,6)]
我自己无法解决这个问题。我知道,a[0:2,0:2]
有效,但在我的代码中,我想使用预定义的1D数组来完成。
由于