Matplotlib 基于手动缩放的 x 轴缩放 y 轴

Question

我有很多在 x 轴上相关的数据，但都以非常不同的 Y 值为中心。 x 上的数据也很长，因此很难看到细节。我希望能够为数据集手动设置 x 轴，然后让绘图根据位于手动设置 x 轴内的数据点的值重新调整 y 轴本身。

matplotlib 可以做到这一点吗？

Answer 1

根据 mdurant 的评论，我写了一些小例子作为你的起点：

import pylab as pl

pl.figure()
x = pl.rand(30)
x.sort()
y =  pl.sin(2 * pl.pi * x) + 0.1 * pl.randn(30)
pl.plot(x, y)

def ondraw(event):
    xlim = pl.gca().get_xlim()
    y_sel = y[pl.logical_and(x > xlim[0], x < xlim[1])]
    pl.gca().set_ylim((y_sel.min(), y_sel.max()))

pl.gcf().canvas.mpl_connect('draw_event', ondraw)

您可能想要添加更多 events 来处理鼠标移动等。

一些cmets：

• 我正在使用pylab，因为它提供了用于生成示例数据的有用功能。但您也可以导入 matplotlib 库。
• ondraw 事件处理程序获取当前坐标区 gca() 的当前 x 限制，选择 y 值以及 x 限制内对应的 x 坐标，并设置由最小和最大 y 值确定的新 y 限制选定的坐标。
• 使用mpl_connect，我们将事件处理程序附加到当前坐标区。每次绘制轴时 - 例如由于新的绘图命令或手动用户交互 - 它调用 ondraw 来更新 y 限制。