我有一個元組串列:
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)]}
有任何想法嗎?
uj5u.com熱心網友回復:
使用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)]}
uj5u.com熱心網友回復:
您可以使用字典.setdefault()來累積結果,使用切片來生成適當的鍵和值。
ans = {}
for entry in res:
ans.setdefault(entry[2:], []).append(entry[:2])
print(ans)
這輸出:
{
(255, 0, 0): [(0, 0), (1, 0), (0, 1), (1, 1)],
(0, 255, 0): [(4, 4), (5, 4), (4, 5), (5, 5)]
}
uj5u.com熱心網友回復:
這個問題有很多更優雅的解決方案,但練習的重點是(可能)讓你練習邏輯和控制流,所以我將展示一個更適合初學者的解決方案:
output = {}
for entry in res:
key, value = entry[2:], entry[:2]
if key not in output:
output[key] = []
output[key].append(value)
正如已經提到collections.defaultdict(list)的那樣,是此類問題的主要候選人。
轉載請註明出處,本文鏈接:https://www.uj5u.com/net/469290.html
上一篇:從具有實際值的嵌套字典中生成對
