I would like the simplest way to an element of a multidimensional array in Python from a tuple/list of the indexes.
I.e. f( M, (a,b,c) ) returns M[a][b][c]
Is there a straightforward way to get this without coding it myself?
Asked
Active
Viewed 418 times
1
-
Can you give some concrete input and output examples – Devesh Kumar Singh Jun 19 '19 at 20:31
-
Is your `M` a list of lists? Did you consider if `numpy.array` would be more suitable for your need? – Imperishable Night Jun 19 '19 at 20:36
-
@ImperishableNight: Yes, M is a list of lists (or tuples) – ThePopMachine Jun 19 '19 at 21:03
-
@ImperishableNight: What changes if it's a numpy.array ? – ThePopMachine Jun 19 '19 at 21:04
-
what can be better than what you wrote? `M[a][b][c]`.... – Tomerikoo Jun 19 '19 at 21:12
-
`numpy.array` supports indexing with tuples, so if `M` is an `array`, `index = (a, b, c); print(M[index])` works. – Imperishable Night Jun 19 '19 at 21:17
-
@Tomerikoo, because you may not know the dimension of the array when your trying to do something general. – ThePopMachine Jun 20 '19 at 01:49
1 Answers
0
You can use numpy arrays for that.
Example:
import numpy as np
x = [ [[1,1,1],[1,1,1], [1,1,1]], [[2,2,2],[2,2,2],[2,2,2]], [[3,3,3], [3,3,3], [3,3,3]]]
matrix = np.array(x)
print(matrix[0,0,0]) # prints first element in matrix
-
Wait, that syntax can't be right, right? Do you mean matrix[(0,0,0)] ? – ThePopMachine Jun 20 '19 at 01:50
-
Both work. You can also use matrix[(0,0,0)]. Just copy the script and test it on python terminal. Remember to install numpy via pip install numpy – Mariano Ezequiel Andres Jun 20 '19 at 19:10