How to slice a 2D array non-consecutively in Python [duplicate]

Question

My friend A is back and she now looks like

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

I need to find the sub-matrix that is H = A[(1,3,7), (2,3,6)]. But thats returns

array([0, 0, 1])

I'm expecting rows 1, 3 and 7 paired with columns 2,3 and 6. I can't seem to find that syntax.

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

akuiper · Accepted Answer · 2017-09-14 19:45:01Z

2

You can use np.ix_:

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

answered Sep 14, 2017 at 19:45

akuiper

216k33 gold badges363 silver badges380 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Brian Dolan Over a year ago

DOH! I forgot about that. Thanks!

Collectives™ on Stack Overflow

How to slice a 2D array non-consecutively in Python [duplicate]

1 Answer 1

1 Comment

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Linked

Related