python - 如何将集合中的元素与多个字典条目相关联

Question

我正在从 XML 文件中提取三个元素的实例:ComponentStr、keyID 和 valueStr。每当我找到 ComponentStr 时，我都想将 keyID:valueStr 添加/关联到它。 ComponentStr 值不是唯一的。当读取多次出现的 ComponentStr 时，我想为该 ComponentStr 组累积 keyID:valueStr。读取 XML 文件后产生的累积数据结构可能如下所示:

组件A:key1:value1, key2:value2, key3:value3

组件B:key4:value4

组件C:key5:value5, key6:value6

生成最终数据结构后，我想对每个 ComponentStr 中的 keyID:valueStr 条目进行排序，并对所有 ComponentStr 进行排序。

我正在尝试在 Python 2 中构建这些数据。ComponentStr 似乎作为一个集合工作得很好。 keyID:valueStr 显然是一个字典。但是，如何将集合中的 ComponentStr 条目与其字典条目相关联？

或者，除了集合和关联的字典条目之外，是否有更好的方法来组织这些数据？每个 keyID 都是唯一的。也许我可以有一个 keyID 的字典:ComponentStr 和 valueStr 的一些组合？构建数据结构后，我可以先根据 ComponentStr 对其进行排序，然后执行某种类型的切片将 keyID:valueStr 分组，然后再根据 keyID 进行排序？看起来很复杂。

Answer 1

一个dict of dicts怎么样？

data = {
'ComponentA': {'key1':'value1', 'key2':'value2', 'key3':'value3'},
'ComponentB': {'key4':'value4'},
'ComponentC': {'key5':'value5', 'key6':'value6'},
}

它维护您的数据结构和映射。有趣的是，dicts 的底层实现类似于 sets 的实现。

这很容易构造成伪代码:

data = {}
for file in files:
    data[get_component(file)] = {}
    for key, value in get_data(file):
        data[get_component(file)][key] = value

在你有重复组件的情况下，你需要将子字典作为默认值，但如果有则添加到前一个。我更喜欢 setdefault 而不是其他解决方案，例如 defaultdict 或使用 __missing__ 子类化 dict，只要我只需要在我的代码中执行一次或两次:

data = {}
for file in files:
    for key, value in get_data(file):
        data.setdefault([get_component(file)], {})[key] = value

它是这样工作的:

>>> d = {}
>>> d.setdefault('foo', {})['bar'] = 'baz'
>>> d
{'foo': {'bar': 'baz'}}
>>> d.setdefault('foo', {})['ni'] = 'ichi'
>>> d
{'foo': {'ni': 'ichi', 'bar': 'baz'}}

或者，当我读到你对另一个答案的评论时说你需要简单的代码，你可以通过一些更冗长和优化程度较低的代码来保持它非常简单:

data = {}
for file in files:
    for key, value in get_data(file):
        if get_component(file) not in data:
            data[get_component(file)] = {}
        data[get_component(file)][key] = value

然后您可以在完成数据收集后进行排序。

for component in sorted(data):
    print(component)
    print('-----')
    for key in sorted(data[component]):
        print(key, data[component][key])