如何将多层嵌套条件的Python循环改写为列表推导式？

你的这段多层嵌套判断代码完全可以优化，既可以改成简洁的列表推导式，也可以用可读性更强的辅助函数实现，两种方式都比原代码更清晰：

方式一：列表推导式 + 链式get()调用 #

利用字典的get()方法（键不存在时返回指定默认值），可以一行完成逻辑，完美替代多层嵌套if：

group_name_list = [
    gn['_source'].get('user', {}).get('target', {}).get('group', {}).get('name', '-')
    for gn in query_res['hits']['hits']
]

原理：每一层get()如果找不到对应键，就返回空字典{}，让后续的get()可以继续调用；最后如果name不存在，直接返回'-'。

方式二：辅助函数（更易维护） #

如果担心列表推导式的链式调用不够直观，或者后续需要扩展判断逻辑，写个辅助函数会更清晰：

def extract_group_name(gn):
    source = gn.get('_source', {})
    user = source.get('user', {})
    target = user.get('target', {})
    group = target.get('group', {})
    return group.get('name', '-')

group_name_list = [extract_group_name(gn) for gn in query_res['hits']['hits']]

这种方式把每一步的取值逻辑拆分，可读性更强，后续修改或调试也更方便。

对比原代码，这两种方式都能得到完全一致的输出，且代码简洁性、可读性都大幅提升，不需要保持原有的嵌套if写法。

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