一个关于python求素数代码的问题

Question

一个python求素数的代码，代码可以运行，输入其他的数字都是正确回答，但是输入9这个数字后就是错误的回答。检查了代码也没发现什么问题。希望有朋友能帮我看看！

def sushu(n):
    for i in range(2,n):
        if n%i == 0:
            return 0
        else:
            return 1

from math import*
n = int(input())
if sushu(n) == 1:
   print("素数")
else:
    print("非素数")

Answer 1

你在循环的时候，只要n除以2不是0就直接返回是素数了
你确定只有9有问题，99没问题？33没问题？只要是奇数你都认为是素数，因为循环永远走不完，第一次就直接return