5.x Python函数和代码复用练习二

连续质数计算

描述

完成以下功能:

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

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23


# 连续输出5个质数
# 先定义质数判定函数
def prime(m):
    for i in range(2, m):
        if m % i == 0:
            return False
        else:
            return True
count = 5
n = eval(input())
N = int(n)
N = N + 1 if N < n else N
while count > 0:
    if prime(N):
        if count > 1:
            print(N, end=',')
        else:
            print(N, end='')
        count -= 1 # 因为要保证输出五个连续的质数，所以只有是质数时count-1，与if平级
    #count -= 1
    #print(count)

    N += 1 # 与第一个if平级