matplotlib交互式图形(在图形上手动绘制线条)

时间:2012-02-03 23:38:08

标签: python matplotlib

我使用matplotlib成功绘制了一组日期排序数据(X轴是日期)。但是,我希望能够手动在绘制的图形上从一个(date1,y1)到另一个(date2,y2)绘制线条。

我似乎无法找到任何显示如何执行此操作的示例 - 或者确实如果它甚至是可行的。

总而言之,这就是我想要做的事情:

  1. 在绘制的图表上绘制一组线
  2. 将手动绘制的行数据保存到文件
  3. 从文件加载手动绘制的线条数据(以重新创建图形)
  4. 理想情况下,我想存储有关绘制线条的“元数据”(例如颜色,线宽等)
  5. 有人可以发布一个骨架片段(最好带有更多信息的链接),以显示我如何开始实现这个(主要要求是能够在图形上手动绘制线条然后保存/加载线条进入情节)。

    注意:通过“手动”,我的意思是能够通过单击某个点来绘制线条,然后单击绘制图形中的另一个点。在两点之间画一条线(或者只是点击一个点并在绘制的图形上的另一个点拖动并释放鼠标)

    [[更新]]

    dawe,非常感谢您提供的代码段。这允许我做我想做的事情 - 但是,一旦在画布上绘制线条(在第二次鼠标点击之后),GUI崩溃并且我在控制台上收到此警告消息:

    /usr/local/lib/python2.6/dist-packages/matplotlib/backend_bases.py:2192: DeprecationWarning: Using default event loop until function specific to this GUI is implemented
      warnings.warn(str,DeprecationWarning)
    

    你知道是什么原因导致了这个警告和突然的程序终止吗?

    此外,是否可以在图表上绘制多条线? (我猜这将涉及编写某种事件处理程序,它将实例化一个linedrawer变量)。目前,在'app'突然终止之前,我有机会只画一行。

1 个答案:

答案 0 :(得分:11)

我会写这样的东西:

import matplotlib.pyplot as plt
class LineDrawer(object):
    lines = []
    def draw_line(self):
        ax = plt.gca()
        xy = plt.ginput(2)

        x = [p[0] for p in xy]
        y = [p[1] for p in xy]
        line = plt.plot(x,y)
        ax.figure.canvas.draw()

        self.lines.append(line)

使用ginput()可以避免更复杂的事件处理。它的“工作方式”是你绘制的东西:

plt.plot([1,2,3,4,5])
ld = LineDrawer()
ld.draw_line() # here you click on the plot

要将行数据保存/加载到文件,您可以使用pickleshelve轻松实现方法。您还可以通过方法draw_line()

传递必要的元数据