首页 文章

如何从matplotlib图中提取数据

提问于
浏览
14

我有一个 wxPython 程序,它从不同的数据集中读取,对数据执行各种类型的简单动态分析,并将数据集的各种组合绘制到 matplotlib canvas . 我希望有机会将当前绘制的数据转储到文件中,以便稍后进行更复杂的分析 .

问题是: matplotlib 中是否有任何方法可以访问 matplotlib.Figure 中当前绘制的数据?

3 回答

  • 1

    Jakub正确地修改了Python脚本,直接从发送到绘图的源中写出数据;这就是我喜欢这样做的方式 . 但作为参考,如果你确实需要从一个情节中获取数据,我认为应该这样做

    gca().get_lines()[n].get_xydata()
    

    或者,您可以单独获取x和y数据集:

    line = gca().get_lines()[n]
    xd = line.get_xdata()
    yd = line.get_ydata()
    
  • 21

    它的Python,因此您可以直接修改源脚本,以便在绘制数据之前将其转储

  • 0

    matplotlib.pyplot.gca 可用于从matplotlib图中提取数据 . 这是一个简单的例子:

    import matplotlib.pyplot as plt
    plt.plot([1,2,3],[4,5,6])
    ax = plt.gca()
    line = ax.lines[0]
    line.get_xydata()
    

    在运行时,您将看到2个输出 - 图表和数据:

    array([[1., 4.],
       [2., 5.],
       [3., 6.]])
    

    enter image description here

    您还可以单独获取x数据和y数据 . 在运行 line.get_xdata() 时,您将获得:

    array([1, 2, 3])
    

    在运行 line.get_ydata() 时,您将获得:

    array([4, 5, 6])
    

    Note: gca 代表获取当前轴

相关问题