一行代码实现素数判断函数

Question

is_prime = /* 你的代码 */

print(is_prime(n)) // n为素数输出True 否则输出False

示例：

fib = lambda n: n if n < 2 else fib(n-1) + fib(n-2)

这个函数可以接受一个整数n作为参数，返回第n个斐波那契数。你可以用print(fib(n))来测试它。

Answer 1

参考答案：

is_prime = lambda n: all(n % i for i in range(2, int(n**0.5)+1)) and n > 1

Comments

赞！