numbers in square pattern 2 in python