python 3 判断是否为素数

#测试过，版本为Python 3.5.0

方法1   for循环版：

import math#需要求平方根
n=int(input('请输入一个正整数\n'))
m=int(math.sqrt(n))

for i in range(2,m+2):
    if n%i==0:
         break
if i==m+1:
     print(n,'是素数')
else:
    print(n,'是合数')

----------------------------分割线--------------------

测试结果：

请输入一个正整数
5
5 是素数

另一个：

请输入一个正整数
15
15 是合数