列表理解语法错误中的嵌套 if / else 子句

Question

从以下列表理解开始：

new_list = [re.sub(r'replace_a', 'with__b', i)
             if not re.findall(r'find_pattern', i, re.M) else 
            re.sub(r'replace_c', r'with__d', i)
            for i in old_list]

现在我想在else 之后添加一个if 条件，它使用re.sub() 替换另一个模式。我试过了：

new_list = [re.sub(r'replace_a', 'with_b', i)
             if not re.findall(r'find_pattern', i, re.M) else
           (re.sub(r'replace_c', r'with_d', i)
             if foo == re.match("pattern", foo))
           for i in old_list]

试图找出使用列表推导的正确方法，但没有运气。

Answer 1

这将更容易作为常规 for 循环阅读：

new_list = []
for i in old_list:
    if not re.findall(r'find_pattern', i, re.M):
        new_list.append(re.sub(r'replace_a', 'with_b', i))
    elif foo == re.match("pattern", foo):
        new_list.append(re.sub(r'replace_c', r'with_d', i))
    # else:
    #    i

您的问题是条件表达式必须始终采用else 子句，因为无论条件是否为真，它都必须具有某些值。但是，对于if 语句，您可以省略else。例如，上面的代码能够使new_list 比old_list 更短，因为不是每个i 都需要调用new_list.append。取消注释 else 的结果与 jpp 的答案相同。

如果您坚持使用列表推导式，那么您可以对其进行格式化以使其更具可读性。考虑

new_list = [re.sub(r'replace_pattern_a', 'with_pattern_b', i)
               if not re.findall(r'find_pattern', i, re.M) else 
            re.sub(r'replace_pattern_c', r'with_pattern_d', i) 
               if foo == re.match("pattern", foo) else
            i
            for i in old_list]

虽然条件表达式确实不是为这种嵌套而设计的。这在视觉上将可以添加到新列表的表达式与用于做出决定的条件分开，但我不是一个大粉丝。还有其他格式可以做出不同的权衡，但 IMO 常规的 for 循环更好。

---

正如 jpp 提到的，另一种重构方法是定义一个生成器函数：

def foo(old_list):
    for i in old_list:
        if not re.findall(r'find_pattern', i, re.M):
            yield re.sub(r'replace_a', 'with_b', i))
        elif foo == re.match("pattern", foo):
            yield re.sub(r'replace_c', r'with_d', i))
        else:
            yield i

new_list = list(foo())

这也有其优点和缺点。 （我认为列举这些可能超出了这个答案的范围，但它确实介于单个列表理解和显式 for 循环的两个极端之间。它也最接近我在 Perl 时代错过的构造， do 语句，类似于 lambda，不带任何参数，但可以包含任意语句。）

Answer 2

当不满足任何条件时，您需要指定else 条件。例如：

import re

old_list = ['a', 'b', 'c']

new_list = [re.sub(r'replace_pattern_a', 'with_pattern_b', i) if not re.findall(r'find_pattern', i, re.M) \
            else re.sub(r'replace_pattern_c', r'with_pattern_d', i) if foo == re.match("pattern", foo) \
            else i for i in old_list]

# ['a', 'b', 'c']

但是，这是不可读的。我建议，如果性能不是问题，您可以创建一个函数来使用标准 if / elif / else 构造提供此操作。