def simple(x:int):
count=0
if x == 1 or x == 0:
return False
for i in range(1,x//2+1):
if x%i==0:
count+=1
if count>1:
return False
return True
k=int(input())
i=0
numb=0
mas=[]
while i<k:
numb+=1
if simple(numb):
i+=1
if simple(i):
ends=numb,i
mas.append((numb,i))
print(mas)
print(ends)
Нужны ли поправки по выводу или сами сделаете?