python判断素数

def f(n):
    if n<2:
        return False
    k = int(n ** 0.5)
    for i in range(2,k+1):
        if n % i == 0:
            return False
    else:
        return True

a=int(input("请输入一个正整数："))
if(f(a)):
    print("{}是素数".format(a))
else:
    print("{}不是素数".format(a))

谢谢大家的支持，您的一键三连是 罡罡同学前进的最大动力！