matplotlibでグラフの右と上の軸を消す

matplotlibで右と上の軸を消すには

ax.spines['right'].set_visible(False)
ax.spines['top'].set_visible(False)
plt.gca().spines['right'].set_visible(False)
plt.gca().spines['top'].set_visible(False)

とすればよい。