7-2 (浙江大学)判断素数

判断一个给定的正整数是否素数

输入格式:
输入在第一行给出一个正整数N（≤ 10），随后N行，每行给出一个小于1000000 的需要判断的正整数

输出格式:
对每个需要判断的正整数，如果它是素数，则在一行中输出Yes，否则输出No

输入样例:
在这里给出一组输入。例如：

2
11
111
结尾无空行
输出样例:
在这里给出相应的输出。例如：

Yes
No
结尾无空行

from math import sqrt
from itertools import count, islice

def prime(n):
    return n > 1 and all(n % i for i in islice(count(2), int(sqrt(n) - 1)))

n = int(input('请输入一个整数：'))
if prime(n):
  print('Yes')
else:
  print('No')

n=eval(input())
for i in range(n):
    a=eval(input())
    f=1
    for j in range(2,a):
        if a%j==0:
            f=0
            break
    if f:
        print('Yes')
    else:
        print('No')