Math 라이브러리를 사용해서 소수를 판별할 수 있지만 itertools 라이브러리를 사용하기 때문에 라이브러리 사용을 줄이고자 sqrt 대신하여 (num//2)+1를 사용하였다.
def prime(num):
if num == 0 or num == 1:
return False
else:
for n in range(2, (num//2)+1):
if num%n == 0:
return False
return True