本文介绍了一种高效处理python三层嵌套字典的方法,该方法能够根据特定条件删除最内层具有相同值的键。 让我们用更简洁的语言重新阐述这个过程,并优化代码。
问题: 如何从三层嵌套字典中删除最内层字典值完全相同的键?
示例:
输入字典 dict1:
立即学习“Python免费学习笔记(深入)”;
dict1 = {'l1':{'pop1':{'a':1}, 'pop2':{'a':1}, 'pop3':{'a':1}}, 'l2':{'pop1':{'b':1}, 'pop2':{'b':1}, 'pop3':{'b':2}}, 'l3':{'pop1':{'c':1}, 'pop2':{'c':2}, 'pop3':{'c':3}}}
目标输出字典 dict2:
dict2 = {'l2':{'pop1':{'b':1}, 'pop2':{'b':1}, 'pop3':{'b':2}}, 'l3':{'pop1':{'c':1}, 'pop2':{'c':2}, 'pop3':{'c':3}}}
解决方案:
以下代码直接操作原字典,避免了不必要的复制,提高了效率:
def remove_replicated(data): for key, value in list(data.items()): # 使用list(data.items())避免修改字典的同时迭代 inner_dicts = list(value.values()) if all(inner_dicts[0] == inner_dict for inner_dict in inner_dicts): del data[key] dict1 = {'l1':{'pop1':{'a':1}, 'pop2':{'a':1}, 'pop3':{'a':1}}, 'l2':{'pop1':{'b':1}, 'pop2':{'b':1}, 'pop3':{'b':2}}, 'l3':{'pop1':{'c':1}, 'pop2':{'c':2}, 'pop3':{'c':3}}} remove_replicated(dict1) print(dict1) # 输出dict2
代码解释:
-
list(data.items()): 为了安全地迭代和修改字典,我们先将字典项转换为列表。直接迭代字典并在迭代过程中修改字典会引发错误。
-
inner_dicts = list(value.values()): 获取当前第二层键的所有内层字典值。
-
if all(inner_dicts[0] == inner_dict for inner_dict in inner_dicts):: 使用all()函数和生成器表达式高效地检查所有内层字典是否相同。
-
del data[key]: 如果所有内层字典相同,则删除对应的第二层键。
这个改进后的代码更简洁、高效,并避免了潜在的错误。 它直接修改输入字典,避免了创建副本带来的额外内存消耗。 这种方法清晰地表达了问题的解决方案,并具有更好的可读性和可维护性。
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END