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