生成器合併排序字典式迭代器

Question

這是Generator to yield gap tuples from zipped iterables上的變體。

我希望設計發電機功能是：

• 接受iterables
• 每個輸入迭代產生零個或多個任意數目的（K，V）中，k不一定是唯一的
• 輸入鍵假定按升序進行排序
• 輸出應該產生(k, (v1, v2, ...))
• 輸出鍵是獨一無二的，應用程序耳以相同的順序與輸入
• 輸出元組的數目等於唯一鍵的輸入
• 的輸出值對應於所有輸入數元組相匹配的輸出密鑰
• 由於輸入和輸出它們可能很大，它們應該被視爲可迭代的對象，並且不會作爲內存中的字典或列表加載。

舉個例子，

i1 = ((2, 'a'), (3, 'b'), (5, 'c')) 
i2 = ((1, 'd'), (2, 'e'), (3, 'f')) 
i3 = ((1, 'g'), (3, 'h'), (5, 'i'), (5, 'j')) 
result = sorted_merge(i1, i2, i3) 
print [result] 

這將輸出：

[(1, ('d', 'g')), (2, ('a', 'e')), (3, ('b', 'f', 'h')), (5, ('c', 'i', 'j'))] 

如果我沒有記錯的話，沒有什麼內置到Python標準庫做到這一點開箱。

Answer 1

雖然沒有一個單一的標準庫函數來完成你想要什麼，有enoughbuildingblocks讓你最方式：

from heapq import merge 
from itertools import groupby 
from operator import itemgetter 

def sorted_merge(*iterables): 
    for key, group in groupby(merge(*iterables), itemgetter(0)): 
     yield key, [pair[1] for pair in group]  

例子：

>>> i1 = ((2, 'a'), (3, 'b'), (5, 'c')) 
>>> i2 = ((1, 'd'), (2, 'e'), (3, 'f')) 
>>> i3 = ((1, 'g'), (3, 'h'), (5, 'i'), (5, 'j')) 
>>> result = sorted_merge(i1, i2, i3) 
>>> list(result) 
[(1, ['d', 'g']), (2, ['a', 'e']), (3, ['b', 'f', 'h']), (5, ['c', 'i', 'j'])] 

注在上面的版本sorted_merge中，爲了產生可讀輸出，我們產生int，list對。如果你想獲得int，<generator>雙反而沒有什麼阻止你改變相關的行

 yield key, (pair[1] for pair in group)   

。

Answer 2

的東西有點不同：

from collections import defaultdict 

def sorted_merged(*items): 
    result = defaultdict(list) 
    for t in items: 
     for k, v in t: 
      result[k].append(v) 
    return sorted(list(result.items())) 

i1 = ((2, 'a'), (3, 'b'), (5, 'c')) 
i2 = ((1, 'd'), (2, 'e'), (3, 'f')) 
i3 = ((1, 'g'), (3, 'h'), (5, 'i'), (5, 'j')) 

result = sorted_merged(i1, i2, i3)