我正在wxPython中编写一个小应用程序,它有一个matplotlib图(使用wxagg后端)面板。我想为用户添加将当前在图中绘制的X,Y数据导出到文本文件的能力。有没有非侵入性的方法来做到这一点?我搜索了很多,似乎找不到任何东西,虽然我觉得它非常简单,就在我面前。
我绝对可以获取数据并将其存储在绘制的某个位置,并使用它 - 但这对于我的代码的较低级别来说是相当具有侵略性的。如果我能做一些像以下那样简单的事情,那将会更容易,更普遍:
x = FigurePanel.axes.GetXData()
y = FigurePanel.axes.GetYData()
希望这有点道理:)
非常感谢!非常感谢任何帮助!
编辑:澄清一下,我想知道怎么做是获取X,Y数据。之后写入文本文件是微不足道的;)
答案 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]))