如何使用 Pandas、Numpy 加速 Python 中的嵌套 for 循环逻辑？

Question

我想看看TestProject表的字段是否包含客户端传过来的参数，嵌套for循环很丑，有没有什么高效简单的实现方法？非常感谢您的任何建议。

def test(parameter_a: list, parameter_b: list) -> bool:
    age = TestPeople.age
    project_code = TestProject.project_code
    if age is not None and age <= 16:
        for code in project_code:
            for p in parameter_a:
                if code[:len(p)] == p:
                    return False
    return True

Answer 1

可以使用itertools.product 和all？

from itertools import product

def test(parameter_a: list, parameter_b: list) -> bool:
    age = TestPeople.age
    project_code = TestProject.project_code
    return (
        age is None 
        or age > 16
        or all(code.startswith(p) for code, p in product(project_code, parameter_a))
    )

解释：

这里python尝试使用short-circuit evaluation，所以会先检查是否age is None，如果是return True，否则检查age > 16，如果是return True，否则检查all是否@ 987654332@对匹配条件code.startswith(p)，如果所有对都满足条件，return True，否则如果其中任何一个对失败，则立即return False而不检查其余的。