写一个判断素数的函数

def my_math(a):

if a==1:

print(“该数不是素数”)

return

elif a==2:

print(“该数是素数”)

return

else:

i=1

while i<=a:

if a%(i+1)==0: #判断一个数不是素数

print(“该数不是素数”)

break

else:

print(“该数是素数”)

break

i+=1

my_math(3)