压缩几个 dicts 或 OrderedDict 保持顺序并提供默认值（如 zip_longest）

Question

我没有找到我要找的答案，所以我自己想出来了，想和你分享。这些：Python equivalent of zip for dictionaries 不实现“默认值”（又名最长风味），也不保持顺序。

如何以“zip_longest”风格在 python 中压缩OrderedDicts？

拥有：

from collections import OrderedDict

o1 = OrderedDict([("key_a", "a1"), ("key_b", "b1")])
o2 = OrderedDict([("key_a", "a2"), ("key_h", "h2")])
o3 = OrderedDict([("key_c", "c3")])
o4 = OrderedDict([("key_x", "x4")])

如何从传递的每个 OrderedDict 中获取成对的键和对应的压缩值，但如果缺少值，则使用 None？

expected_result = [
    ('key_a', ('a1', 'a2', None, None)),
    ('key_b', ('b1', None, None, None)),
    ('key_h', (None, 'h2', None, None)),
    ('key_c', (None, None, 'c3', None)),
    ('key_x', (None, None, None, 'x4')),
]

Answer 1

您可以使用列表推导式迭代从给定 dicts 的所有键创建的 OrderedDict 并输出键值元组，其值是具有生成器表达式的元组，该生成器表达式遍历 dicts 并返回每个字典中的给定键：

def zip_dicts(*dicts):
    return [(k, tuple(d.get(k) for d in dicts)) for k in OrderedDict.fromkeys(k for d in dicts for k in d)]

这样zip_dicts(o1, o2, o3, o4) 返回：

[('key_a', ('a1', 'a2', None, None)),
 ('key_b', ('b1', None, None, None)),
 ('key_h', (None, 'h2', None, None)),
 ('key_c', (None, None, 'c3', None)),
 ('key_x', (None, None, None, 'x4'))]

Answer 2

@blhsing 提供了最可爱的solution。这是另一种选择。 在 python 2 和 3 中工作：

def zip_ordered_mappings_default(*mappings):
    collected_keys = set()
    for key in (key for mapping in mappings for key in mapping):
        if key not in collected_keys:
            collected_keys.add(key)
            yield key, tuple(collection.get(key) for collection in mappings)

# usage:
assert list(zip_ordered_mappings_default(o1, o2, o3, o4)) == expected_result

结果项的顺序保持不变，相当于从传递的 dict 键创建的有序集的并集。 IE。新键出现在末尾。

该解决方案非常通用，也可以与常规dicts 一起使用：

assert dict(zip_ordered_mappings_default(dict(o1), dict(o2), dict(o3), dict(o4))) == {
    'key_a': ('a1', 'a2', None, None),
    'key_b': ('b1', None, None, None),
    'key_h': (None, 'h2', None, None),
    'key_c': (None, None, 'c3', None),
    'key_x': (None, None, None, 'x4')
}

当然，在 dicts 的情况下，项目的顺序不会保留，但结果值中的元组反映了函数参数的顺序。