matplotlibでのグラフの形状の変え方
グラフの描画範囲`ax`を正方形にする

ax=plt.subplot(aspect='equal')