Python: Return 2 ints for index in 2D lists given item

Asked 2011-04-25T05:05:37.240

I've been tinkering in python this week and I got stuck on something.
If I had a 2D list like this:

myList = [[1,2],[3,4],[5,6]]

and I did this

>>>myList.index([3,4])

it would return

However, I want the index of something in side one of the lists, like this

    >>>myList.index(3)

and it would return

1, 0

Is there anything that can do this?

Cheers

python nested-lists

Edit Report

1 Answer

Using simple genexpr:

def index2d(list2d, value):
    return next((i, j) for i, lst in enumerate(list2d) 
                for j, x in enumerate(lst) if x == value)

print index2d([[1,2],[3,4],[5,6]], 3)
# -> (1, 0)

answered 2011-04-25T06:52:32.503