python编写一个函数,参数为整数,判断参数是否为质数,并设计主函数测试
时间: 2023-05-15 19:02:36 浏览: 109
Python编写一个函数,可以判断传入的整数是否为质数。质数是指只能够被1和自身整除的正整数。当传入的整数可以被其他整数整除时,即不是质数。
以下是判断质数的函数:
```python
def is_prime(num):
if num < 2:
return False
for i in range(2, int(num ** 0.5) + 1):
if num % i == 0:
return False
return True
```
函数中的变量`num`即为传入的需要判断的整数,首先进行一个判断,如果小于2的话直接返回False。然后进行一个循环,从2开始,一直到该数的平方根,如果能够整除则返回False,否则返回True。
接下来就需要设计一个主函数来测试该函数是否可以正常使用,具体代码如下:
```python
def main():
num = int(input("请输入需要判断的整数: "))
if is_prime(num):
print(num, "是质数")
else:
print(num, "不是质数")
if __name__ == '__main__':
main()
```
在主函数中,首先需要用户输入一个整数作为参数,然后调用判断质数的函数进行判断,如果是质数则输出这是质数,否则输出不是质数。
最后需要注意的是,在主函数中可以使用`__name__`特殊变量来判断该Python文件是否直接被运行,如果是则执行主函数,如果被导入则不执行。
阅读全文