是否有用于评估短路函数/表达式列表的Python习惯用法？

Question

我写了一个简单的脚本来解决一个"逻辑谜题",即从学校获得许多规则的拼图类型,然后必须能够找到问题的解决方案,例如"有五位音乐家名为A,B,C ,D和E在音乐会上演奏,每个演奏一个接一个...如果A在B之前,D不在最后......谁在什么时候演奏？等等

为了评估可能的解决方案,我将每个"规则"编写为一个单独的函数,它将评估一个可能的解决方案(简单地表示为字符串列表)是否有效,例如

#Fifth slot must be B or D
def rule1(solution):
    return solution[4] == 'B' or solution[4] == 'D'

#There must be at least two spots between A and B
def rule2(solution):
    returns abs(solution.index('A') - solution.index('B')) >= 2

#etc...

我有兴趣找到Pythonic方法来测试一个可能的解决方案是否通过所有这些规则,并能够在第一个失败后停止评估规则.

起初我写了最简单的事情:

def is_valid(solution):
    return rule1(solution) and rule2(solution) and rule3(solution) and ...

但这看起来相当丑陋.我想也许我可以通过列表理解之类的东西让这个阅读更加优雅......

def is_valid(solution)
    rules = [rule1, rule2, rule3, rule4, ... ]
    return all([r(solution) for f in rules])

...但后来我意识到,由于列表理解是在all()评估函数之前生成的,因此它具有不会短路的副作用 - 即使第一个返回,每个规则也会被评估False.

所以我的问题是:是否有更多的Pythonic /功能方式能够评估True/ False表达式列表,具有短路,而不需要写出长列表return f1(s) and f2(s) and f3(s) ...？

Answer 1

使用生成器表达式:

rules = [ rule1, rule2, rule3, rule4, ... ]
rules_generator = ( r( solution ) for r in rules )
return all( rules_generator )

语法糖:你可以省略额外的括号:

rules = [ rule1, rule2, rule3, rule4, ... ]
return all( r( solution ) for r in rules )

生成器(基本上)是一个带有.next()方法的对象,它返回一些可迭代的下一个项目.这意味着他们可以做一些有用的事情,比如以块的形式读取文件而不将其全部加载到内存中,或者迭代到大整数.你可以for透明地循环遍历它们; Python在幕后处理它.例如,range是Py3k中的生成器.

您可以使用yield语句而不是return函数定义来滚动自己的自定义生成器表达式:

def integers():
    i = 0
    while True:
        yield i

Python将处理保存函数的状态等.他们太棒了!