Python如何将含嵌套循环与条件判断的代码改写为列表推导式

你写的列表推导式报错核心有两个问题：

• 结构错位：Python列表推导式的if/else三元表达式必须写在元素生成位置，不能拆开放在两个for循环语句中间，你把分支判断插在了迭代子句的位置，解析器无法识别语法。
• 迭代顺序错误：原嵌套逻辑是外层先遍历sorting，内层才遍历timeframe，你把timeframe的遍历写在了sorting外层，就算语法通过，运行逻辑也和原代码完全不一致。

这类带分支、分支下嵌套不同层数循环的逻辑完全可以改写成列表推导式，核心思路是把每个sorting对应的分支产出，封装成可迭代对象，再在外层推导式里统一展开即可，代码和原逻辑100%等价：

self.array_out += [
    entry
    for sorting in self.possible_sortings
    if self.sorting_flags[sorting]
    for entry in (
        # 支持时间维度筛选的分支：遍历所有符合标记的timeframe生成条目
        ([subreddit_name, sorting, tf, ammount_posts] for tf in self.possible_timeframes if self.timeframe_flags[tf])
        if self.sorting_timeframe_possible_flag[sorting]
        # 不支持时间维度筛选的分支：仅生成1条timeframe为None的条目
        else ([subreddit_name, sorting, None, ammount_posts],)
    )
]

• 迭代顺序完全对齐原代码：最外层先遍历所有sorting规则，先过滤掉未开启标记的sorting项，和原逻辑一致。
• 对每个有效sorting，通过三元判断选择对应的可迭代对象：支持timeframe时用生成器产出所有符合条件的timeframe条目；不支持时用单元素元组产出固定的None条目，外层推导式会自动把这些可迭代对象里的元素逐个展开追加到结果数组。
• 性能相比原循环确实有小幅提升，减少了反复调用append方法的开销，没有额外的逻辑判断成本。
• 注意如果后续分支逻辑继续复杂化，不建议强行用列表推导式实现，拆回普通for循环的可读性会更高。

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