Как сохранить информацию о ax.plot в виде данных пикселей?

Question

Учитывая простой пример кода, в котором dr aws строки между заданными координатами с использованием ax.plot ()

import matplotlib.pyplot as plt
x_values = [0, 2, 4, 5]
y_values = [3, 1, 5, 2]
fig, ax = plt.subplots(1, 1)
ax.set_xlim(0,5)
ax.set_ylim(0,5)
ax.plot(x_values, y_values)
plt.show()

Я видел , есть ax.imshow(), , но это не вернет матрицу / вектор. Есть у кого-нибудь идеи?

Answer 1

Вы можете сохранить изображение как jpg, а затем открыть его как массив numpy. Вы можете проверить форму, чтобы убедиться, что она вам нравится.

import numpy as np 
from PIL import Image

i = Image.open('name of your jpg.jpg')
array = np.asarray(i)
print(array.shape)

Если вы хотите изменить форму:

array.flatten().reshape(#M dim, #N dim)