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()我确信它更适合。