您可以根据另一个图的轴范围为一个图设置新的轴范围。在两个轴上使用 xlim_changed 事件调用一个函数,该函数根据当前限制调整另一个绘图的限制。
需要确保在更改限制之前断开事件的连接,以免陷入无限循环。
下面是一个实现,其中底部图与顶部图相比移动了 100 个单位。
import numpy as np; np.random.seed(1)
import matplotlib.pyplot as plt
x = np.linspace(0,500,1001)
y = np.convolve(np.ones(20), np.cumsum(np.random.randn(len(x))), mode="same")
fig, (ax, ax2) = plt.subplots(nrows=2)
ax.set_title("original axes")
ax.plot(x,y)
ax2.set_title("offset axes")
ax2.plot(x,y)
offset = lambda x: x + 100
inverse_offset = lambda x: x - 100
class OffsetAxes():
def __init__(self, ax, ax2, func, invfunc):
self.ax = ax
self.ax2 = ax2
self.func = func
self.invfunc = invfunc
self.cid = ax.callbacks.connect('xlim_changed', self.on_lims)
self.cid2 = ax2.callbacks.connect('xlim_changed', self.on_lims)
self.offsetaxes(ax, ax2, func)
def offsetaxes(self,axes_to_keep, axes_to_change, func):
self.ax.callbacks.disconnect(self.cid)
self.ax2.callbacks.disconnect(self.cid2)
xlim = np.array(axes_to_keep.get_xlim())
axes_to_change.set_xlim(func(xlim))
self.cid = ax.callbacks.connect('xlim_changed', self.on_lims)
self.cid2 = ax2.callbacks.connect('xlim_changed', self.on_lims)
def on_lims(self,axes):
print "xlim"
if axes == self.ax:
self.offsetaxes(self.ax, self.ax2, self.func)
if axes == self.ax2:
self.offsetaxes(self.ax2, self.ax, self.invfunc)
o = OffsetAxes(ax, ax2, offset, inverse_offset)
plt.show()