using index() on multidimensional lists

前端 未结 8 2081
北恋
北恋 2020-12-10 03:00

For a one dimensional list, the index of an item is found as follows:

 a_list = [\'a\', \'b\', \'new\', \'mpilgrim\', \'new\']
 a_list.index(\'mpilgrim\')
         


        
8条回答
  •  爱一瞬间的悲伤
    2020-12-10 03:38

    For two dimensional list; you can iterate over rows and using .index function for looking for item:

    def find(l, elem):
        for row, i in enumerate(l):
            try:
                column = i.index(elem)
            except ValueError:
                continue
            return row, column
        return -1
    
    tl = [[1,2,3],[4,5,6],[7,8,9]]
    
    print(find(tl, 6)) # (1,2)
    print(find(tl, 1)) # (0,0)
    print(find(tl, 9)) # (2,2)
    print(find(tl, 12)) # -1
    

提交回复
热议问题