CODEWITHSUNDEEP
python
Pmsmm
Pmsmm

Reputation: 440

How to index into nested lists?

I want to make a function in python which receives as arguments a matrix, a coordinate for the line and another coordinate for the column. For example: A matrix m=[[1,2,3], [4,5,6]] and the function will receive the arguments (m,0,0) It should return 1 (Which is the number located in position 0,0 in the matrix).

Upvotes: 1

Views: 5543

Answers (1)

Erica
Erica

Reputation: 2457

Think of it as a list of lists rather than a "matrix", and the logic becomes more obvious. Matrix m has two elements: m[0] = [1, 2, 3] and m[1] = [4, 5, 6]. So accessing a single value from within those lists requires another index. For example, m[0][1] = 2.

def matrix(m, a, b):
    return m[a][b] # element b from list a in list m

If you really want to use a matrix, consider numpy.

Upvotes: 6

Related Questions