소수구하기. 에라토스테네스의 체

import sys
import math

# 1부터 n까지의 수 중에서 소수 찾기
n=int(input())
sosu=[True for i in range(n+1)] # 모든 숫자가 소수라고 가정

for i in range(2, int(math.sqrt(n))+1):
    if sosu[i]==True:
        for j in range(i+i,n+1,i):
            sosu[j]=False
            
print([i for i in range(2,n+1) if sosu[i]==True])