Извлечение данных из рисунка 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.]))