如何在Python中非连续地切片二维数组[重复]

Question

我的朋友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]])