自定义函数处理素数

解题思路:

注意事项:

参考代码:

def function(x):
    for i in range(2, x):
        if x % i == 0:
            return print("not prime")
            break
    else:
        return print("prime")

if __name__ == '__main__':
    n = int(input())
    function(n)