По данному числу k найдите k-е по счету простое число

# -*- coding: utf-8 -*-
import math
import random
 
 
def is_prime(n):
    """https://stackoverflow.com/questions/36095518/"""
    if n % 2 == 0 and n > 2:
        return False
    return all(n % i for i in range(3, int(math.sqrt(n)) + 1, 2))
 
 
k = 100
number = 2
i = 0
while True:
    if is_prime(number):
        i += 1
        print('%d) %d' % (i, number))
        if i == k:
            break
    number += 1
print(number)

import math
 
#Введите нужное значение k
k = 5
 
numb_prost = 0
found = False
 
def find_prost(n):
 
    for i in range(2, round(math.sqrt(n)+1)):
        if (n%i==0):
            #print("i=",i)
 
            return "Непростое"
 
    return "Простое"
 
 
n = 1
 
if k==1:
    print("Ответ задачи равен 2")
else:
 
    while (found == False):
   
        n += 1
        answ = find_prost(n)
        #print("Число n ",n," ", answ)
 
        if answ == "Простое":
            numb_prost += 1
 
        if numb_prost==k:
            found = True 
 
    print("Ответ задачи=", n)

k = int(input())
if k == 1:
    print(2)
    quit()
 
a = k * 17
prime = [True] * (a + 1)
prime[0] = prime[1] = False
for i in range(2, a + 1):
    if not prime[i]:
        continue
    for j in range(i * i, a + 1, i):
        prime[j] = False
 
count = 0
for i in range(a):
    if prime[i]:
        count += 1
    if count == k:
        print(i)
        quit()