matplotlibで正方形の領域に描画

ここからコピペしただけ。これ以外上手くいかなかった。
http://atztogo.hatenablog.com/archive/2014/7

xlim = ax.get_xlim() 
ylim = ax.get_ylim() 
aspect = (xlim[1] - xlim[0]) / (ylim[1] - ylim[0]) 
ax.set_aspect(aspect)
axはsubplotの返り値。