python - 扁平化 OrderedDict 值的联合

Question

我正在寻找一种 Pythonic 方式来获取 OrderedDict 的 .values 的并集，在 折叠组成值的列表之后。我希望结果是一维列表，根据自定义顺序排序。

我在做什么:

from collections import OrderedDict    
order = list('acebd')
d = OrderedDict([(1, ['a', 'e']),
                 (2, ['a', 'b', 'c'])
                ])

和期望的结果:

['a', 'c', 'e', 'b'] # ordered according to `order`, where len(order)
                     # may be greater than len(result) here

我目前的实现

def flatten(iterable):
    """Flatten a nested iterable.  Not my code."""
    it = iter(iterable)

    for e in it:
        if isinstance(e, (list, tuple)):
            for f in flatten(e):
                yield f
        else:
            yield e

print(list(set(flatten(d.values()))))
['a', 'e', 'b', 'c']

但不确定如何映射到排序，并寻找更优雅或更快的解决方案。

Answer 1

你不需要展平，它们只是简单的列表，只需将它们 itertools.chain 放在一起即可。
您可以使用集合作为列表推导中的守卫来获得您想要的顺序:

In []:
import itertools as it
s = set(it.chain.from_iterable(d.values()))
[o for o in order if o in s]

Out[]:
['a', 'c', 'e', 'b']