Python 代码优化：判断一个数是否为质数

这段代码旨在判断一个数是否为质数，存在以下问题：

1. 在第6行中，将'math.sqrt(a)'作为'range()'函数的参数，但'range()'函数的参数必须为整数，因此需要将其转换为整数。
2. 在第6行中，应该使用'int(math.sqrt(a) + 1)'而不是'math.sqrt(a) + 1'，因为'range()'函数的结束值是不包含在范围内的。
3. 在第8行中，应该使用'print('no')'而不是'print('yes')'，因为在循环中找到一个可以整除的数时，应该输出'no'并立即退出循环。

以下为优化后的代码：

import math
a = int(input())
if a == 1:
    print('no')
else:
    for i in range(2, int(math.sqrt(a)) + 1):
        if a % i == 0:
            print('no')
            break
    else:
        print('yes')

优化后的代码修复了上述问题，并使用'else'语句来判断循环是否正常结束，提高代码的可读性。