如何在不使用函数代码的情况下在python中找到第n个素数？

import math

def is_prime(num):
    if num < 2:
        return False
    for i in range(2, int(math.sqrt(num)) + 1):
        if num % i == 0:
            return False
    return True

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

n = int(input("请输入要找到的第n个素数："))
result = find_nth_prime(n)
print("第{}个素数是：{}".format(n, result))