输入两个数判断是否是素数如果是就求和

Question

输入两个数判断是否是素数，如果是就求和，有一个不是就重新输入

Answer 1

# # -*- coding: utf-8 -*-

def f(n):
    # 判断数是否为素数
    if n<2:
        return False
    k = int(n ** 0.5)
    for i in range(2,k+1):
        if n % i == 0:
            return False
    else:
        return True


a,b = input('请输入两个整数(空格隔开)：').split()

while not f(int(a)) or not f(int(b)):
    a, b = input('请输入两个整数(空格隔开)：').split()

print('两个数的和为：',int(a)+int(b))