在列表列表中找到项目的索引

Question

我有一个清单清单：

colours = [["#660000","#863030","#ba4a4a","#de7e7e","#ffaaaa"],["#a34b00","#d46200","#ff7a04","#ff9b42","#fec28d"],["#dfd248","#fff224","#eefd5d","#f5ff92","#f9ffbf"],["#006600","#308630","#4aba4a","#7ede7e","#aaffaa"]]

搜索列表并返回其中一项的位置的最干净方法是什么"#660000"？

我已经看过该index方法，但是似乎并没有在列表中解压缩列表。

postion = colours.index("#660000")

给出：ValueError: ['#660000'] is not in list，不是[0][0]我所期望的…

Answer 1

我会做这样的事情：

[(i, colour.index(c))
 for i, colour in enumerate(colours)
 if c in colour]

这将返回一个元组列表，其中第一个索引是第一个列表中的位置，第二个索引是第二个列表中的位置（注意：c您要查找的颜色，即"#660000"）。

对于问题中的示例，返回值为：

[(0, 0)]

如果您只需要懒惰地找到找到颜色的第一个位置，则可以使用以下方法：

next(((i, colour.index(c))
      for i, colour in enumerate(colours)
      if c in colour),
     None)

这将返回找到的第一个元素的元组，或者None如果找不到元素（您也可以删除None上面的参数，StopIteration如果找不到元素，则将引发异常）。

编辑：正如@RikPoggi正确指出的那样，如果匹配数很高，这将带来一些开销，因为colour要对其进行两次迭代查找c。我认为这对于少量匹配并在单个表达式中给出答案是合理的。但是，为避免这种情况，您还可以使用相同的思想定义方法，如下所示：

def find(c):
    for i, colour in enumerate(colours):
        try:
            j = colour.index(c)
        except ValueError:
            continue
        yield i, j

matches = [match for match in find('#660000')]

请注意，由于find是生成器，因此您可以像上面的示例中那样实际使用它，next以在第一个比赛时停止并跳过进一步的查找。