从matplotlib图中检索XY数据

时间:2013-11-21 20:01:22

标签: python matplotlib wxpython

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

我绝对可以获取数据并将其存储在绘制的某个位置,并使用它 - 但这对于我的代码的较低级别来说是相当具有侵略性的。如果我能做一些像以下那样简单的事情,那将会更容易,更普遍:

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

希望这有点道理:)

非常感谢!非常感谢任何帮助!

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

1 个答案:

答案 0 :(得分:22)

这有效:

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.]])

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

编辑: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]))