matplotlib:更改当前轴实例(即gca())

时间:2013-10-28 00:44:13

标签: python python-2.7 matplotlib

我使用了draw a colorbar whose height matches the master axes的技巧。代码就像

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable
import numpy as np

ax = plt.subplot(111)
im = ax.imshow(np.arange(100).reshape((10,10)))

# create an axes on the right side of ax. The width of cax will be 5%
# of ax and the padding between cax and ax will be fixed at 0.05 inch.
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%", pad=0.05)

plt.colorbar(im, cax=cax)

这个技巧很好用。但是,由于附加了新轴,因此图形的当前实例变为cax - 附加轴。因此,如果执行类似

的操作
plt.text(0,0,'whatever')

文本将在cax而不是ax上绘制 - 即im所属的轴。

同时,gcf()。轴显示两个轴。

我的问题是:如何使当前轴实例(由gca()返回)成为im所属的原始轴。

1 个答案:

答案 0 :(得分:56)

使用plt.sca(ax)设置当前轴,其中ax是您要激活的Axes对象。