从matplotlib图中检索XY数据

Question

我在wxPython中编写了一个小应用程序,它有一个matplotlib图(使用wxagg后端)面板.我想为用户添加将当前在图中绘制的X,Y数据导出到文本文件的能力.有没有非侵入性的方法来做到这一点？我搜索了很多,似乎找不到任何东西,虽然我觉得它非常简单,就在我面前.

我可以肯定地获取数据并将其存储在绘制时的某个位置,并使用它 - 但这将是相当侵入性的,进入我的代码的较低级别.如果我能做一些像以下那样简单的事情,那将会更容易,更普遍:

x = FigurePanel.axes.GetXData()
y = FigurePanel.axes.GetYData()

希望这有点意义:)

非常感谢!任何帮助是极大的赞赏!

编辑:澄清,我想知道怎么做是获取X,Y数据.之后写入文本文件是微不足道的;)

Answer 1

这有效:

In [1]: import matplotlib.pyplot as plt

In [2]: plt.plot([1,2,3],[4,5,6])
Out[2]: [<matplotlib.lines.Line2D at 0x30b2b10>]

In [3]: ax = plt.gca() # get axis handle

In [4]: line = ax.lines[0] # get the first line, there might be more

In [5]: line.get_xdata()
Out[5]: array([1, 2, 3])

In [6]: line.get_ydata()
Out[6]: array([4, 5, 6])

In [7]: line.get_xydata()
Out[7]: 
array([[ 1.,  4.],
       [ 2.,  5.],
       [ 3.,  6.]])

我通过在轴对象中挖掘来找到它们.我只能找到一些关于这些函数的最小信息,显然你可以给它们一个布尔标志来获取原始或处理过的数据,不知道它的含义.

编辑:Joe Kington展示了一种稍微简洁的方法:

In [1]: import matplotlib.pyplot as plt

In [2]: lines = plt.plot([1,2,3],[4,5,6],[7,8],[9,10])

In [3]: lines[0].get_data()
Out[3]: (array([1, 2, 3]), array([4, 5, 6]))

In [4]: lines[1].get_data()
Out[4]: (array([7, 8]), array([ 9, 10]))