如何使用嵌套列表的第一个元素打印字典键

Question

我有一个字典，其值是嵌套列表。例如：

d = {1: [[2, 5], [3, 4]]}

如何使用嵌套列表的第一个元素创建链接键并打印

[[1, 2], [1, 3]]

Answer 1

你可以使用简单的列表理解，比如这个

>>> d = {1: [[2, 5], [3, 4]]}
>>> [[key, value[0]] for key in d for value in d[key]]
[[1, 2], [1, 3]]

这可以理解，就像这样

>>> result = []
>>> for key in d:
...     for value in d[key]:
...         result.append([key, value[0]])
... 
>>> result
[[1, 2], [1, 3]]

或dict.items，

>>> result = []
>>> for key, values in d.items():
...     for value in values:
...         result.append([key, value[0]])
...         
>>> result
[[1, 2], [1, 3]]

主要思想是，我们使用for key in d迭代字典的键，并且对于每个键，我们使用d[key]获取与其对应的值并迭代它的元素，以准备结果为[key, value[0]]。