python - 无需 lambda 即可减少

Question

我成功编写了一些代码，如果列表中的所有项目都是素数，则该代码返回 true。
我认为这是 reduce 的一个很好的候选者 - 到目前为止我只将 reduce 与 lambda 结合使用 - 是否可以避免 lambda 并使用直接函数？

def is_prime(list):
    np = "Prime"
    for n in list:
        if n < 2:
            np = "NonPrime"
            print np
            # return np
        else:
            for i in range(3, n):   #int(math.sqrt(n))
                if n % i == 0:
                    np ="NonPrime"
                    print np
    return np   

Answer 1

您可能想要重构此代码以检查数字是否为素数，然后使用它来测试数字列表是否为 all() 素数。 all() 会比 reduce() 更好，因为它会短路。

注意:您不应该使用 list 作为变量或参数，因为它隐藏了 python 内置类型。

def is_prime(n):
    if n < 2:
        return False
    for i in range(3, n):   #int(math.sqrt(n))
        if n % i == 0:
            return False
    return True

def all_primes(iterable):
    return all(is_prime(n) for n in iterable)

>>> all_primes([2,3,5,7])
True
>>> all_primes([2,3,5,8])
False

如果您确实想将其实现为 reduce() 那么，您可以使用 and 将 bool 值列表有效地减少为单个 bool 值，有一个运算符and 在 operator 模块中定义，operator.and_:

from functools import reduce   # Py3
import operator as op

def all_primes(iterable):
    return reduce(op.and_, (is_prime(n) for n in iterable))