Python中，输入一个整数，求第n小的质数代码怎么写

以下是 Python 代码，用来求第 n 小的质数：

def is_prime(num):
    '''判断一个数是否为质数'''
    if num <= 1:
        return False
    for i in range(2, int(num ** 0.5) + 1):
        if num % i == 0:
            return False
    return True

def nth_prime(n):
    '''获取第 n 小的质数'''
    count = 0
    num = 2
    while count < n:
        if is_prime(num):
            count += 1
        num += 1
    return num - 1

对于输入的整数 n，调用 nth_prime(n) 函数即可得到第 n 小的质数。

注意：如果质数范围较大，上面的代码可能会执行很慢。实际应用中，应该采用更高效的算法，例如 Sieve of Eratosthenes。