accessing an element in a numpy matrix using a list (python)

Question

I have a 10x10 numpy matrix and I have a list containing indices of elements I want to query.

matrix = np.zeros((10, 10), dtype=int)

indices = [[2,3], [3,4]]

the issue I'm facing is, what I actually want is the element matrix[2,3] but matrix[indices[0]] gives a different output because the latter actually means matrix[[2,3]]. the output is 2 different rows.

How do I get around this problem?

arr[ [2,3], [3,4] ] or indices=([2,3], [3,4]) and arr[indices]. numpy.org/doc/stable/reference/… — hpaulj
– hpaulj, Commented Jun 5, 2021 at 17:19

Abhi Minhas · Accepted Answer · 2021-06-05 17:25:13Z

1

I ended up finding a solution.

storing the indices as tuples inside of a list works

indices = [(2,3), (3,4)] and then matrix[indices[0]] gives the desired output

answered Jun 5, 2021 at 17:25

Abhi Minhas

371 silver badge7 bronze badges

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

Collectives™ on Stack Overflow

accessing an element in a numpy matrix using a list (python)

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