python素数判断

素数判断

import math
def is_prim(num):           #判断素数
    is_prime = True
    for i in range(2,int(math.sqrt(num))+1):
        if num%i==0:
            is_prime = False
            break
    if is_prime and num>1:
        return True
    else:
        return False

>>> is_prim(17)
True
>>> is_prim(18)
False