python - 需要从 python 列表理解中提取多个值？

Question

我有一本带有元组键的字典，如下所示:

(1111, 3454): 34.55555
(1123, 4665): 67.12
(1111, 9797): 5.09

我需要进行列表理解，以获取具有匹配的第一个元素的所有条目的值。问题是，我还需要元组的第二个值...

interimlist = [v for k,v in mydict.items() if k[0]==item[0]]

如果元组的第一个元素正确(项目是迭代器变量)，那么我现在就可以提取值。我希望输出是(值，第二个元组编号)的元组列表，因此通过示例点，如果 item[0] 为 1111，则输出将是以下输出:

[(34.55555, 3454), (5.09, 9797)]

Answer 1

这里的字典没有以良好的结构存储。必须迭代所有键/​​值才能进行一次查找，因此检索的时间复杂度为 O(n)。

您应该对数据进行一次重新键入，在字典中添加另一层嵌套:

>>> d
{(1111, 3454): 34.55555, (1123, 4665): 67.12, (1111, 9797): 5.09}
>>> d_new = {}
>>> for (k1, k2), v in d.items():
...     if k1 not in d_new:
...         d_new[k1] = {}
...     d_new[k1][k2] = v

现在，O(1) 查找已恢复:

>>> d_new[1111]
{3454: 34.55555, 9797: 5.09}
>>> [item[::-1] for item in d_new[1111].items()]
[(34.55555, 3454), (5.09, 9797)]