Python新手求助：如何避免丑陋的嵌套for循环？

嘿，我完全懂这种三层嵌套循环看着乱糟糟的感觉！其实有几个实用的方法能让你的代码更清爽、可读性更强，我给你分享几个常用方案：

方案1：用生成器函数封装遍历逻辑 #

把多层遍历的细节封装到一个生成器函数里，主代码只需要处理扁平化后的每个条目，这样逻辑拆分更清晰：

def get_ward_details(data):
    for province in data:
        province_name = province["name"]
        for district in province["districts"]:
            district_name = district["name"]
            for ward in district["wards"]:
                # 把需要的信息以元组形式返回
                yield ward["name"], district_name, province_name

# 主代码变得非常简洁
for ward_name, district_name, province_name in get_ward_details(data):
    # 执行你的业务函数
    print(f"{ward_name}, {district_name}, {province_name}")

这个方法的优势在于：

• 把遍历逻辑和业务逻辑分离，代码更模块化
• 生成器是惰性求值的，不会一次性加载所有数据到内存里，适合处理大数据量
• 后续如果要调整遍历规则（比如过滤某些省/区），只需要修改生成器函数即可

方案2：用itertools.chain扁平化嵌套结构 #

如果你不想单独写函数，可以用itertools.chain.from_iterable结合生成器表达式来一次性扁平化所有层级：

from itertools import chain

# 逐层展开嵌套结构，同时绑定对应的省、区名称
flattened_wards = chain.from_iterable(
    ( (ward["name"], district["name"], province["name"]) for ward in district["wards"] )
    for province in data
    for district in province["districts"]
)

for ward_info in flattened_wards:
    print(f"{ward_info[0]}, {ward_info[1]}, {ward_info[2]}")

这种写法更紧凑，但如果嵌套层级更多或者逻辑更复杂，可读性会不如生成器函数。

方案3：列表推导式（适合小数据量） #

如果你的数据量不大，也可以用列表推导式一次性生成所有需要的信息列表，再遍历处理：

ward_info_list = [
    (ward["name"], district["name"], province["name"])
    for province in data
    for district in province["districts"]
    for ward in district["wards"]
]

for ward_name, district_name, province_name in ward_info_list:
    print(f"{ward_name}, {district_name}, {province_name}")

注意：列表推导式会把所有数据一次性加载到内存中，数据量大时不推荐使用。

总结 #

个人最推荐生成器函数的方案，它兼顾了代码可读性、内存效率和可维护性，非常适合处理这种多层嵌套的数据结构。

内容的提问来源于stack exchange，提问作者Nguyen Anh Tan