You can use list comprehensions:
[[1 for i in range(square)] for i in range(square)]
Or this will probably work, too:
[[1] * square for i in range(square)]
As an extra point, if you want to loop over a piece of code n times, its usually better to use a for-loop, rather than a while-loop:
for i in range(n):
<code>