結果 : how to write n square in python