python - 是否有一种优雅的替代方法可以在不使用大量 if 语句的情况下过滤这些结果？

Question

我正在 python 中执行一些模拟，生成 1000 个结果。每个结果都具有风险、成本等多个属性

我现在想确定满足特定条件的结果，例如:

Factor 1, cost should be between 10 and 20
Factor 2, risk should be between 0 and 5
Factor 3...
Factor 4...
....

目前我正在使用一系列嵌套的 if 语句来执行此操作。随着更多因素的添加，嵌套变得困惑。是否有一种优雅的替代方法可以根据特定条件进行过滤？

Answer 1

有很多解决方案取决于您的数据和偏好的代码风格。例如:

>>> conditions = (lambda x: 10 < x.cost < 20, lambda x: 0 < x.risk < 10)
>>> filter(lambda x: all(cond(x) for cond in conditions), result)

或者只是:

>>> conditions = lambda x: 10 < x.cost < 20 and 0 < x.risk < 10
>>> filter(conditions, result)

或者:

>>> [r for r in result if 10 < r.cost < 20 and 0 < r.risk < 10]