Python:如何使用两个1D数组下标2D数组

时间:2018-04-09 16:16:57

标签: python arrays numpy

我是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)]。相反,我的bc是2x3矩阵:b=c=[(1, 2, 3), (4,5,6)]

我自己无法解决这个问题。我知道,a[0:2,0:2]有效,但在我的代码中,我想使用预定义的1D数组来完成。

由于

0 个答案:

没有答案