使用X和Y值索引嵌套列表

Question

我有一个顶点列表和它们之间的长度。我想创建一个函数，通过搜索列表来查找列表中两个顶点之间的距离。

名单看​​起来像：

lengths = [[X, Y, length], [X, Z, length], [Y, Z, length], etc] 

说顶点是X和Y我能指标采用X列表，Y找到X和Y之间的长度？

目前，我这样做是这样的：

def find_sides(X, Y, lengths): 
    for a in lengths: 
     if a[0] == X and a[1] == Y: 
      length = a[2] 
    return length 

但由于长度的名单不断增加，这可能需要一段时间。

感谢

Answer 1

您可以使用字典：

lengths = {(X, Y): length, (X, Z): length, (Y, Z): length, ...} 

然后，它只是一个索引字典的事：

length = lengths[(X, Y)] 

您可以将您的当前目录与拟议的格式词典理解：

lengthDict = {(a, b): c for [a, b, c] in lengthList}