如何让matplotlib打开一个用户评论框?

时间:2017-05-15 07:33:40

标签: python python-2.7 matplotlib

我有一个交互式情节,可以收听某些按键和点击,但我希望用户能够添加评论。我知道艺术家事件通常不允许这样做(他们正在收听个人印刷!但是我可以让matplotlib打开一个新的窗口,其中有一个小的“插入注释”区域吗?理想情况下,窗口退出并返回到主窗口(原始) )当用户点击返回时窗口。

import numpy as np
import matplotlib.pyplot as plt

def onpick(event):
    ''' '''
    if event.mouseevent.button == 1: #only want lmb clicks
        selection = event.artist
        xdata = selection.get_xdata()
        ydata = selection.get_ydata()
        ind = event.ind
        point = tuple(zip(xdata[ind], ydata[ind]))
        xclick,yclick = point[0] 

        print('[x,y]=',xclick,yclick)   

def on_key(event):
    ''' 
        Handles predefined key-press events 
    ''' 
    print('Key press:\'%s\'' %(event.key))
    if event.key == ' ': #spacebar
        print 'Space'
        #do a thing
    if event.key == 'e': 
        print 'eeeeee'
        #do another thing
    if event.key == 'C':
        print 'How do make a comment. ...'
        comment = 'Whatever the user entered'
        return comment

        # when done return

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, facecolor='#FFFFCC')

x, y = 4*(np.random.rand(2, 100) - .5)
ax.plot(x, y, 'o', picker = 6)
ax.set_xlim(-2, 2)
ax.set_ylim(-2, 2)

keyID = fig.canvas.mpl_connect('key_press_event', on_key)
clickID = fig.canvas.mpl_connect('pick_event', onpick)

plt.show()

1 个答案:

答案 0 :(得分:1)

Matplotlib 即将现在推出了一个TextBox Widget。在this example中查看其用法。

或者,您可以使用Tkinter tkSimpleDialog向用户请求评论。

w = tkSimpleDialog.askstring("Title", "Please type comment")

然后,您可以使用评论注释最后选取的点。

完整示例(在python 2.7中运行):

import numpy as np
import matplotlib.pyplot as plt
import Tkinter, tkSimpleDialog

xy = [(0,0)]

def onpick(event):
    ''' '''
    if event.mouseevent.button == 1: #only want lmb clicks
        selection = event.artist
        xdata = selection.get_xdata()
        ydata = selection.get_ydata()
        ind = event.ind
        point = tuple(zip(xdata[ind], ydata[ind]))
        xclick,yclick = point[0] 
        xy[0] = (xclick,yclick)
        print('[x,y]=',xclick,yclick)   

def on_key(event):
    print('Key press:\'%s\'' %(event.key))
    if event.key == 'c':
        root = Tkinter.Tk()
        root.withdraw()
        w = tkSimpleDialog.askstring("Title", "Please type comment")
        if w != None:
            ax.annotate(w, xy=xy[0], xytext=(20,-20), 
                    arrowprops=dict(facecolor='black', width=2, headwidth=6),
                    textcoords='offset points')
            ax.figure.canvas.draw_idle()

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111, facecolor='#FFFFCC')

x, y = 4*(np.random.rand(2, 100) - .5)
ax.plot(x, y, 'o', picker = 6)
ax.set_xlim(-2, 2)
ax.set_ylim(-2, 2)

keyID = fig.canvas.mpl_connect('key_press_event', on_key)
clickID = fig.canvas.mpl_connect('pick_event', onpick)

plt.show()

enter image description here