python输入一个数n如何判断是否为素数

import math

m = int(input("请输入一个整数(>1)："))

k = int(math.sqrt(m))

flag = True #假设输入的整数为素数

i = 2

while (i <= k and flag == True):

  if (m % i == 0): flag = False #可以整除，肯定不是素数，结束循环

  else: i += 1

if (flag == True): print(m, "是素数！")

else: print(m, "是合数！")

import math;n=0

for m in range(101,201,2):

  k = int(math.sqrt(m))

  for i in range(2,k+2):

    if m % i == 0:break

  if i == k+1:

    if n % 10 == 0:print()

    print('%d' % m,end=' ')

    n += 1