使用matplotlib和twinx进行光标跟踪

时间:2013-05-21 14:27:04

标签: python matplotlib

我想同时跟踪两个轴上数据坐标的鼠标坐标。我可以相对于一个轴跟踪鼠标位置就好了。 问题是:当我使用twinx()添加第二个轴时,两个Cursors仅报告与第二个轴相关的数据坐标。

例如,我的游标(fernmuffy)报告y - 值为7.93

Fern: (1597.63, 7.93)
Muffy: (1597.63, 7.93)

如果我使用:

    inv = ax.transData.inverted()
    x, y = inv.transform((event.x, event.y))

我得到了一个IndexError。

所以问题是:如何修改代码以跟踪两个轴的数据坐标?


enter image description here

import numpy as np
import matplotlib.pyplot as plt
import logging
logger = logging.getLogger(__name__)

class Cursor(object):
    def __init__(self, ax, name):
        self.ax = ax
        self.name = name
        plt.connect('motion_notify_event', self)

    def __call__(self, event):
        x, y = event.xdata, event.ydata
        ax = self.ax
        # inv = ax.transData.inverted()
        # x, y = inv.transform((event.x, event.y))
        logger.debug('{n}: ({x:0.2f}, {y:0.2f})'.format(n=self.name,x=x,y=y))


logging.basicConfig(level=logging.DEBUG,
                    format='%(message)s',)
fig, ax = plt.subplots()

x = np.linspace(1000, 2000, 500)
y = 100*np.sin(20*np.pi*(x-1500)/2000.0)
fern = Cursor(ax, 'Fern')
ax.plot(x,y)
ax2 = ax.twinx()
z = x/200.0
muffy = Cursor(ax2, 'Muffy')
ax2.semilogy(x,z)
plt.show()

2 个答案:

答案 0 :(得分:3)

由于回调的工作方式,事件总是在顶轴返回。您只需要一些逻辑来检查事件是否发生在我们想要的轴上:

class Cursor(object):
    def __init__(self, ax, x, y, name):
        self.ax = ax
        self.name = name
        plt.connect('motion_notify_event', self)

    def __call__(self, event):
        if event.inaxes is None:
            return
        ax = self.ax
        if ax != event.inaxes:
            inv = ax.transData.inverted()
            x, y = inv.transform(np.array((event.x, event.y)).reshape(1, 2)).ravel()
        elif ax == event.inaxes:
            x, y = event.xdata, event.ydata
        else:
            return
        logger.debug('{n}: ({x:0.2f}, {y:0.2f})'.format(n=self.name,x=x,y=y))

这可能是变换堆栈中的一个微妙的错误(或者这是正确的用法,并且幸运的是它之前使用了元组),但无论如何,这将使它工作。问题是transform.py中第1996行的代码需要返回2D ndarray,但身份转换只返回传递给它的元组,这就是产生错误的原因。

答案 1 :(得分:0)

您可以通过这种方式使用一个光标(或事件处理程序)跟踪两个轴坐标:

import numpy as np
import matplotlib.pyplot as plt
import logging
logger = logging.getLogger(__name__)

class Cursor(object):
    def __init__(self):
        plt.connect('motion_notify_event', self)

    def __call__(self, event):
        if event.inaxes is None:
            return
        x, y1 = ax1.transData.inverted().transform((event.x,event.y))
        x, y2 = ax2.transData.inverted().transform((event.x,event.y))
        logger.debug('(x,y1,y2)=({x:0.2f}, {y1:0.2f}, {y2:0.2f})'.format(x=x,y1=y1,y2=y2))

logging.basicConfig(level=logging.DEBUG,
                    format='%(message)s',)
fig, ax1 = plt.subplots()

x = np.linspace(1000, 2000, 500)
y = 100*np.sin(20*np.pi*(x-1500)/2000.0)
fern = Cursor()
ax1.plot(x,y)
ax2 = ax1.twinx()
z = x/200.0
ax2.plot(x,z)
plt.show()

(我得到了#34;太多的索引和#34;当我像OP一样使用ax2.semilogy(x,z)时,但没有解决这个问题。)

ax1.transData.inverted().transform((event.x,event.y))代码执行从指定轴上的显示到数据坐标的转换,并且可以随意使用任一轴。