結果 : code to make a square in python