python求第n小的质数

输入一个正整数n,求第n小的质数。
输入一个不超过10000的正整数n,输出第n小的质数。
这个怎么写啊

img

主要是要写一个判断一个数是否是指数的函数,然后从2开始逐步增加待定数, 望采纳

def is_prime(num):
    # 判断是否是质数
    if num < 2:
        return False
    elif num == 2:
        return True
    elif num % 2 == 0:
        return False
    else:
        for i in range(3, int(num**0.5)+1, 2):
            if num % i == 0:
                return False
        return True

def find_nth_prime(n):
    count = 0
    guess = 2
    while count < n:
        if is_prime(guess):
            count += 1
        guess += 1
    return guess - 1

# 主程序开始
n = int(input("请输入一个不超过10000的正整数:"))
result = find_nth_prime(n)
print("第", n, "小的质数为:", result)