如何获得列表中的列表索引?

时间:2013-03-05 20:45:47

标签: python

list = ["word1", "word2", "word3"]
print list.index("word1")

这很好用!

但我如何得到这个的索引:

list = [["word1", "word2", "word3"],["word4", "word5", "word6"]]
print list.index("word4")

那不起作用,错误:

ValueError: "word4" is not in list

我希望得到像1,0

这样的答案

5 个答案:

答案 0 :(得分:3)

尝试这样的事情:

def deep_index(lst, w):
    return [(i, sub.index(w)) for (i, sub) in enumerate(lst) if w in sub]

my_list = [["word1", "word2", "word3"],["word4", "word5", "word6"]]
print deep_index(my_list, "word4")
>>> [(1, 0)]

这将返回一个元组列表,第一个元素指向外部列表中的索引,第二个元素指向该子列表中单词的索引。

答案 1 :(得分:2)

对于多维索引,假设您的数据可以表示为NxM(而不是列表的一般列表),numpy非常有用(并且快速)。

import numpy as np
list = [["word1", "word2", "word3"],["word4", "word5", "word6"]]
arr = np.array(list)
(arr == "word4").nonzero()
# output: (array([1]), array([0]))
zip(*((arr == "word4").nonzero()))
# output: [(1, 0)] -- this gives you a list of all the indexes which hold "word4"

答案 2 :(得分:1)

我认为你必须手动找到它 -

def index_in_list_of_lists(list_of_lists, value):
   for i, lst in enumerate(list_of_lists):
      if value in lst:
         break
   else:
      raise ValueError, "%s not in list_of_lists" %value

   return (i, lst.index(value))


list_of_lists = [["word1", "word2", "word3"],["word4", "word5", "word6"]]
print index_in_list_of_lists(list_of_lists, 'word4') #(1, 0)

答案 3 :(得分:1)

def get_index(my_list, value):
    for i, element in enumerate(my_list):
        if value in element:
            return (i, element.index(value))
    return None


my_list= [["word1", "word2", "word3"], ["word4", "word5", "word6"]]
print get_index(my_list, "word4")

打印(1,0)

答案 4 :(得分:1)

将来,请尽量避免命名变量list,因为它会覆盖Python的内置list

lst = [["word1", "word2", "word3"],["word4", "word5", "word6"]]

def find_index_of(lst, value):
   for index, word in enumerate(lst):
      try:
        inner_index = word.index(value)
        return (index, inner_index)
      except ValueError:
        pass
   return ()

这循环遍历lst的每个元素,它将:

  • 尝试获取index的{​​{1}}。如果我们找到了元素,那么让我们返回索引。
  • 但是,如果Python抛出value(因为该元素不在列表中),那么让我们继续下一个列表。
  • 如果找不到任何内容,请返回空元组。

输出:

ValueError