python画正方形的代码