可以循環通過多個不同大小的Python字典

Question

我從另一個類

>>> print init_treats 
{('001', 0): init_treat_001_0, ('001', 3): init_treat_001_3, ('001', 2): init_treat_001_2, ('002', 0): init_treat_002_0, ('002', 1): init_treat_002_1, ('002', 2): init_treat_002_2, ('002', 3): init_treat_002_3, ('001', 1): init_treat_001_1} 

>>> print init_untreats 
{'002': init_untreat_002, '001': init_untreat_001} 

我怎樣才能生成以下繼承以下兩個Python字典？

init_treat_001_0 + init_treat_001_1 + init_treat_001_2 + init_treat_001_3 + 
init_untreat_001 == 0 


init_treat_002_0 + init_treat_002_1 + init_treat_002_2 + init_treat_002_3 + 
init_untreat_002 == 0 

Answer 1

排序您init_treats鍵：

treats = sorted(init_treats) 

現在你可以使用itertools.groupby()將它們分組對你的關鍵的第一部分：

from itertools import groupby 
from operator import itemgetter 

for untreat, group in groupby(sorted(init_treats), itemgetter(0)): 
    # group is now a sorted iterator of keys with the same first value 
    if init_untreat[untreat] + sum(map(init_treats.get, group)) == 0: 
     # sum of init_treat_n_m + init_untreat_n is 0 

由於這種使用排序，這是一個O（NlogN）解決方案（N是init_treats字典的大小）

你可以用字典查一個O（N + K）溶液（K爲init_untreats字典的大小）：

sums = init_untreat.copy() 
for untreat, id in init_treats: 
    sums[untreat] += init_treats[untreat, id] 

for untreat, total in sums.items(): # use sums.iteritems() in Python 2 
    if total == 0: 
     # sum of init_treat_n_m + init_untreat_n is 0 

因爲K是在你的情況下N多始終較小，漸進地說，這是一個O（N）算法，當然。