How to play a list that skips the numbers according to the number typed

Question

Preferably something that explains the answer below (without the spaces)

0

1

4

9

16

Answer 1

You can use list comprehension for this question, much more idiomatic:

[ i**2 for i in range(5) ]

Answer 2

If you want something that generates this result that you showed in the question, just do it:

i = 0
lista=[]

while i < 5:
   lista.append((i ** 2)) 
   i += 1

Output:

[0, 1, 4, 9, 16]

and so on.