Skip to main content
 首页 » 编程设计

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

2024年08月27日43zhenyulu

我正在从 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 进行排序?看起来很复杂。

请您参考如下方法:

一个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])