python - 将值从列表列表添加到字典键

Question

我有以下列表和字典:

list = [[1,2,2],[2,3,3],[3,4,4]
dict = {1:[], 2:[], 3:[]}

假设我想将 dict[keys] 中的所有值附加到各自的键中，这样:

dict = {1:[2,2], 2:[3,3], 3:[4,4]}

我尝试使用多个循环来完成此任务。然而，每当我这样做时，所有值都会被同时搜索并附加到每个 dict[key] 中。任何帮助将不胜感激!

Answer 1

l = [[1,2,2],[2,3,3],[3,4,4]]

# last key wins 
d = {sublist[0]:sublist[1:] for sublist in l}

print(d)

您可以根据列表元素构建字典。不要使用 list 或 dict 作为名称，它们会隐藏内置函数。

sublist[0] 是第一个元素，sublist[1:] 是 l 中每一项的其余元素。

有关此处使用的语法的更多信息:PEP 274: Dict Comprehensions

输出:

{1: [2, 2], 2: [3, 3], 3: [4, 4]}     # last key wins (no duplicates present)

---

如果您也需要合并，可以使用:

# merging keys into values as list of sublists
l = [[2,2,2],[2,3,3],[3,4,4]]
d1 = {}
for sub in l:
    # creates dict of { key: [ [int,...], [int,...] ], ... } for same keys
    d1.setdefault(sub[0],[]).append(sub[1:]) 

或者通过展平:

# merging keys into values as flattened list
d2 = {}
for sub in l:
    # creates dict of { key: [ int,...,int,... ], ... } for same keys
    d2.setdefault(sub[0],[]).extend(sub[1:]) 

输出:

{2: [[2, 2], [3, 3]], 3: [[4, 4]]}    # merging keys into values as list of sublists

{2: [2, 2, 3, 3], 3: [4, 4]}          # merging keys into values as flattened list

为了合并，我使用了 dict.setdefault()您可能想看一下 collections.defaultdict()我确信它更适合。