Разница между переменной и входом в python3?

Question

Я написал следующее:

i = 0 
n = int(input(5)) 
while i <= (n-1): 
  print(i**2)
  i +=1

Я ожидал следующий результат:

0
1
4
9
16

Но, похоже, я получу этот результат, только если напишу код следующим образом:

i = 0
n = 5 
while i <= (n-1):
  print(i**2)
  i +=1

Может кто-нибудь посоветовать, пожалуйста?