Python实现分段函数计算、素数判断及结果分类输出

Python等级考试五级编程题：分段函数的计算及取值。

编写程序实现以下功能：

1. 连续输入5个x值，按给定分段函数计算对应的y值，将结果转换为整数后添加到列表a中；
2. 判断列表a中5个元素是否为素数：非素数存入临时列表a1，素数存入临时列表a2；
3. 将a1、a2分别转换为元组b1、b2，分别输出b1、b2的元素及对应个数。

分段函数定义：

• x < 0时，y = 0
• 0 ≤ x < 5时，y = x
• 5 ≤ x < 10时，y = 3x - 5
• 10 ≤ x < 20时，y = 0.5x - 2
• x ≥ 20时，y = 2x

现有待补全代码如下：

a = []
a1 = []  # 暂时存储非素数的值
a2 = []  # 暂时存储素数的值
b1 = ()
b2 = ()
for i in range(5):
    x = int(input("输入x="))
    if x < 0:
        y = 0
    elif 0 <= x < 5:
        y = x
    elif 5 <= x < 10:
        y = 3 * x - 5
    elif 10 <= x < 20:
        y = 0.5 * x - 2
    elif x >= 20:
        y = 2 * x
    ①

for i in a:
    if ②:
        a1.append(i)
    else:
        flag = True
        for j in range(2, i):
            if i % j == 0:
                ③
                break
        if ④:
            a2.append(i)
        else:
            a1.append(i)
b1 = tuple(a1)
b2 = tuple(a2)
print("非素数的个数有" + str(len(b1)) + "个,", "非素数为：", b1)
print("素数的个数有" + str(len(b2)) + "个,", "素数为：", b2)