python从matplotlib点击返回ydata

时间:2014-09-27 00:38:16

标签: python matplotlib

我试图点击matplotlib中的一个图并将点击位置的y值返回给变量。我使用了matplotlib中的例子,它从event.ydata打印出位置,但无法弄清楚如何返回它!

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(np.random.rand(10))

def onclick(event):
    print 'button=%d, x=%d, y=%d, xdata=%f, ydata=%f'%(event.button, event.x, event.y, event.xdata, event.ydata)

cid = fig.canvas.mpl_connect('button_press_event', onclick)

1 个答案:

答案 0 :(得分:1)

如果我理解正确,你想要弹出一个情节并等待用户点击其中的一个点然后再继续。据我所知,没有预先打包的方式,但下面的代码应该做你想要的。完全披露:它确实使用全局变量。

import matplotlib.pyplot as plt
import numpy as np
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(np.random.rand(10))

retval = -1
def onclick(event):
    print 'button=%d, x=%d, y=%d, xdata=%f, ydata=%f'%(event.button, event.x, event.y, event.xdata, event.ydata)
    # Record the x location of the user's click in the global variable and close the figure
    global retval
    retval = event.xdata
    plt.close()

cid = fig.canvas.mpl_connect('button_press_event', onclick)
# Bring up the figure (and wait)
plt.show()

print 'User selected point {0}'.format( retval )