python - 如何在Python中非连续地切片二维数组

Question

这个问题已经有答案了:

Selecting specific rows and columns from NumPy array (4 个回答)

已关闭 5 年前。

我的 friend A回来了，她现在看起来像

A = np.array([
    [0,1,1,1,0,0,0,0], 
    [1,0,0,1,0,0,0,0],
    [1,0,0,1,0,0,0,0],
    [1,1,1,0,0,0,0,0],
    [0,0,0,1,0,1,0,0],
    [0,0,0,0,1,0,1,1],
    [0,0,0,0,0,1,0,1],
    [0,0,0,0,0,1,1,0],
             ])

我需要找到子矩阵H = A[(1,3,7), (2,3,6)] 。但这就是返回

array([0, 0, 1])

我期待第 1、3 和 7 行与第 2,3 和 6 列配对。我似乎找不到该语法。

H = [[0,1,0],
     [1,0,0],
     [0,0,1]]

Answer 1

您可以使用np.ix_ :

A[np.ix_((1,3,7),(2,3,6))]
#array([[0, 1, 0],
#       [1, 0, 0],
#       [0, 0, 1]])