Írj egy függvényt, amely eldönti egy adott számról, hogy prím vagy sem!
szam = 29
def prim_e(szam):
    if szam < 2:
        return False
    for i in range(2, int(szam**0.5) + 1):
        if szam % i == 0:
            return False
    return True