代码问题-使用生成器扁平化字典

nested_dict = {
    'a': 1,
    'b': {
        'c': 2,
        'd': {
            'e': 3,
            'f': 4
        }
    },
    'g': 5
}

def flatten_dict(d, parent_key='', sep='_'):
    for k, v in d.items():
        new_key = f"{parent_key}{sep}{k}" if parent_key else k
        if isinstance(v, dict):
            yield from flatten_dict(v, new_key, sep=sep)
        else:
            yield new_key, v

# 使用生成器扁平化字典
flattened = dict(flatten_dict(nested_dict))
print(flattened)

{
    'a': 1,
    'b_c': 2,
    'b_d_e': 3,
    'b_d_f': 4,
    'g': 5
}

def flatten_dict_iterative(d, sep='_'):
    stack = [((), d)]
    while stack:
        path, current = stack.pop()
        for k, v in current.items():
            new_path = path + (k,)
            if isinstance(v, dict):
                stack.append((new_path, v))
            else:
                yield sep.join(new_path), v

# 使用迭代方法扁平化字典
flattened_iterative = dict(flatten_dict_iterative(nested_dict))
print(flattened_iterative)

{
    'a': 1,
    'b_c': 2,
    'b_d_e': 3,
    'b_d_f': 4,
    'g': 5
}