educoder实训——函数【2】

素数判定

任务描述

本关任务：编写一个能进行素数判定的小程序。

相关知识
为了完成本关任务，你需要掌握：

素数判断
素数判断
编程实现判定用户输入的一个正整数 n 是否为素数。

将素数的判定代码定义为一个函数，接受用户输入的正整数 n，返回 n 是否是素数，n 为素数时返回 True，不是素数时返回 False。

并调用该函数判定用户输入的一个正整数 n 是否为素数。

编程要求

根据提示，在右侧编辑器补充代码，进行素数判定的小程序。

测试说明
平台会对你编写的代码进行测试：

输入格式:
输入一个正整数

输出格式:
若该数为素数，输出“X是素数”
若该数为非素数，输出“X不是素数”
其中X为输入的整数

测试输入：5；
预期输出：
5是素数

代码

def is_prime(n): 
    "