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

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] 获取与其对应的值code> 并迭代其中的元素，以准备 [key, value[0]] 形式的结果。