Довольно просто!
def prime(x): if x == 1: return False else: for a in range(2,x): if x % a == 0: return False return True