Numpy slicing by array?

Question

import numpy as np
a = np.array([4, 3, 2, 1])
b = np.array([1, 2, 4, 3])
c = np.stack((a, b))
i = np.array([0, 1])
print(c[i])

I got:

[[4 3 2 1]
 [1 2 4 3]]

but my expectd output:

[4 2]

How can I implement this?

@hilberts_drinking_problem that's exactly what I want, thanks a lot. — Zacchaeus
– Zacchaeus, Commented Oct 12, 2020 at 10:27

Pluto · Accepted Answer · 2020-10-12 23:05:09Z

1

What do you want to achieve here?

c = [[4 3 2 1] [1 2 4 3]]; from np.stack call

[c[0][0],c[1][1]] returns [4 2] which is what you want.

Pluto

8831 gold badge10 silver badges28 bronze badges

answered Oct 12, 2020 at 10:15

Mel

112 bronze badges

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

1 Answer 1