使用default dict创建树数据类型：如何检测特定条目并执行操作

嘿，很高兴看到你尝试用defaultdict构建嵌套树结构来替代类——这种场景下确实会比多层类更灵活，也能避免类嵌套带来的混淆！针对你需要检测特定条目并执行操作的需求，我给你几个实用的方案：

方案1：递归遍历（适配任意深度嵌套） #

这种方法最通用，不管你的树结构嵌套多深，都能遍历到每一个条目，找到目标后执行自定义操作。

首先我们先构建一个模拟的嵌套defaultdict结构（贴合你的数据存储场景）：

from collections import defaultdict
import functools

# 用partial快速创建支持无限嵌套的defaultdict
nested_dict = functools.partial(defaultdict, lambda: defaultdict(dict))

# 示例数据：模拟传感器、执行器的多层数据
data = nested_dict()
data['sensor']['temp']['room1'] = 25.6
data['sensor']['humidity']['room1'] = 45
data['actuator']['light']['room2'] = 'on'
data['actuator']['fan']['room3'] = 'off'

接下来写一个递归遍历函数，专门用来检测目标条目并执行操作：

def traverse_and_operate(node, target_key, operation):
    # 检查当前节点是否是字典类型（包括defaultdict）
    if isinstance(node, dict):
        for key, value in node.items():
            # 匹配到目标条目，触发自定义操作
            if key == target_key:
                operation(key, value, node)  # 传入node方便修改值
            # 递归遍历子节点，深入嵌套结构
            traverse_and_operate(value, target_key, operation)

# 定义你想要执行的操作，比如打印条目信息，或者修改值
def my_operation(key, value, parent_node):
    print(f"找到目标条目：{key} = {value}")
    # 举个修改值的例子：如果是温度数据，就加0.3
    if key == 'temp' and isinstance(value, dict):
        parent_node[key]['room1'] += 0.3
        print(f"已更新温度值：{parent_node[key]['room1']}")

# 调用函数，查找所有名为'temp'的条目并执行操作
traverse_and_operate(data, 'temp', my_operation)

方案2：直接路径访问（适合结构固定的场景） #

如果你的嵌套结构相对固定，知道目标条目的大致路径，那直接通过键路径访问会更高效，不用遍历整个树：

# 比如我们知道目标在sensor -> humidity下
if 'humidity' in data['sensor']:
    target_data = data['sensor']['humidity']
    print(f"湿度数据：{target_data}")
    # 执行操作，比如修改room1的湿度值
    data['sensor']['humidity']['room1'] = 47

一些实用小贴士 #

• 如果需要检测的是特定值而非键，只需要把递归函数里的判断条件改成if value == target_value即可
• 如果你担心嵌套过深导致递归栈溢出，可以把递归改成迭代方式（用栈或者队列来存储待遍历的节点）
• 要是需要批量检测多个目标条目，可以把target_key改成一个列表，遍历的时候判断key是否在列表里

内容的提问来源于stack exchange，提问作者Mike