从 Matplotlib 图中提取数据

Question

我有一个简单的图形，我使用 matplotlib.plot 函数来绘制它。 例如：

有没有办法提取数据点并将其（如在 matlab 中）粘贴到我不知道的 Excel 表中？ 我想假设许多数字是随机创建的，在看到结果之前我不知道我需要哪些数据/数字。

Answer 1

要提取数据点，您可以为绘图分配一个变量：

graph = plt.plot(your_data)
data_points = graph[0].get_data()

提取线图的数据点并将它们保存到 csv 文件的示例：

In[1]:  import matplotlib.pyplot as plt
        import numpy as np

        x = np.linspace(-1, 1, 5)
        y = 2*x + 1
        xy = plt.plot(x, y)
        data = xy[0].get_data()
        print(data)
        plt.show()

        with open('data.csv', 'w') as myfile: 
            writer = csv.writer(myfile)
            writer.writerow(['x', 'y'])
            for i in range(len(data[0])): 
                writer.writerow([data[0][i], data[1][i]])            
Out[1]: (array([-1. , -0.5,  0. ,  0.5,  1. ]), array([-1.,  0.,  1.,  2.,  3.]))