python绘制正方形代码