Swift含闭包的isPrime函数转Python 3.7.4单行实现求助

没问题，我帮你把这个Swift的单行素数判断函数完美复刻到Python里～

先拆解下你原来Swift函数的核心逻辑：

func isPrime(_ number: Int) -> Bool { return number > 1 && !(2..<number).contains {number % $0 == 0} }

它做了两件事：

• 首先确保数字大于1（素数的基本条件）
• 然后检查从2到number-1的范围内，没有任何一个数能整除它（通过取反contains的结果实现，因为contains会在找到一个因数时返回true，取反后就代表不存在因数）

在Python里，我们可以用all()函数来实现“所有元素都满足条件”的判断，正好对应取反后的contains逻辑。直接写成单行返回的函数如下：

def isPrime(number: int) -> bool:
    return number > 1 and all(number % i != 0 for i in range(2, number))

代码对应解释： #

• number > 1 和Swift里的条件完全一致
• all(number % i != 0 for i in range(2, number))：这里的生成器表达式会遍历range(2, number)（对应Swift的2..<number）里的每个数，检查number是否不能被它整除。all()会在所有检查都通过时返回true，和你Swift代码里!(contains { ... })的效果完全等价。

如果想要优化性能，其实可以只检查到int(number**0.5) +1（因为如果一个数有因数，那它的平方根以内必然存在一个因数），优化后的单行版本是：

def isPrime(number: int) -> bool:
    return number > 1 and all(number % i != 0 for i in range(2, int(number**0.5) + 1))

内容的提问来源于stack exchange，提问作者Les Churchman