How do I create a multi-dimensional index array?

Question

I want to create a 3d array where basically the content is identical to the indeces used to access it. So m[2,5] would result in array([2, 5]). I couldn't find an obvious solution with the numpy functions indices, ogrid, concatenate, etc.

At the moment I'm using this, but was wondering whether there is a solution that makes better use of the API:

a, b = 3, 4
m = np.ones((a, b, 2))
for x in range(a):
    m[x,:, 1] = np.array(range(b))
for y in range(b):
    m[:,y,0] = np.array(range(a))

Quang Hoang · Accepted Answer · 2020-11-12 15:18:05Z

1

Try np.mgrid:

a, b = 3, 4
m = np.mgrid[:a,:b].transpose(1,2,0)

print(m[1,2])
# array([1, 2])

answered Nov 12, 2020 at 15:18

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

Collectives™ on Stack Overflow

How do I create a multi-dimensional index array?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related