连续质数计算

gzj137070928

获得用户输入数字N，计算并输出从N开始的5个质数，单行输出，质数间用逗号,分割。

注意：需要考虑用户输入的数字N可能是浮点数，应对输入取整数；最后一个输出后不用逗号。

def prime(n):
    s = 0
    for i in range(2,n-1):
        if n % i == 0:
            s =1
    if s == 0:
        return n
    else:
        return 0

n = eval(input())
flag = 5
num = []
while flag:
    if prime(n):
        if flag > 1:
            print(n, end="\033[31:1m,\033[0m")
        else:
            print(n)
        flag -= 1
    n += 1

参考示例：

def prime(m):
    for i in range(2,m):
        if m % i == 0:
            return False
    return True

n = eval(input())
n_ = int(n)
n_ = n_+1 if n_ < n else n_
count = 5

while count > 0:
    if prime(n_):
        if count > 1:
            print(n_, end=",")
        else:
            print(n_, end="")
        count -= 1 
    n_ += 1