洛谷p5723质数口袋

洛谷p5723质数口袋是一道关于质数的题目，题目描述如下：

小A有一个质数口袋，里面可以装各个质数。他从2开始，依次判断各个自然数是不是质数，如果是质数就会把这个数字装入口袋。现在小A想知道，他口袋里的第n个质数是多少，请你编写一个程序帮助他。

输入格式：
输入一个整数n。

输出格式：
输出一个整数，表示小A口袋里的第n个质数。

注意：本题数据范围是1≤n≤10000。

以下是一个Python的解法：

n = int(input())
cnt = 0
i = 2
while True:
    flag = True
    for j in range(2, int(i ** 0.5) + 1):
        if i % j == 0:
            flag = False
            break
    if flag:
        cnt += 1
        if cnt == n:
            print(i)
            break
    i += 1