以列表作为值的字典

Question

我有一个元组列表：

res=[(0, 0, 255, 0, 0),(1, 0, 255, 0, 0),(0, 1, 255, 0, 0),(1, 1, 255, 0, 0),
(4, 4, 0, 255, 0),(5, 4, 0, 255, 0),(4, 5, 0, 255, 0),(5, 5, 0, 255, 0)]

这是我的想法：

keys = [l[2:] for l in res]
values = [l[:2] for l in res]
d=dict(zip(keys, values))

这是我的输出：

{(255, 0, 0): (1, 1), (0, 255, 0): (5, 5)}

我的输出是错误的，我需要这个：

{(255, 0, 0): [(0, 0),(1,0),(0,1),(1,1)], 
(0, 255, 0): [(4,4),(5,4),(4,5),(5,5)]}

有任何想法吗？

Answer 1

使用collections.defaultdict：

from collections import defaultdict 

out = defaultdict(list)

for t in res:
    out[t[2:]].append(t[:2])

dict(out)

或者用古典词典：

out = {}

for t in res:
    k = t[2:]
    if k not in out:
        out[k] = []
    out[k].append(t[:2])

输出：

{(255, 0, 0): [(0, 0), (1, 0), (0, 1), (1, 1)],
 (0, 255, 0): [(4, 4), (5, 4), (4, 5), (5, 5)]}