Вывод графика Matplotlib в виде массива оттенков серого

Question

Мне нужно вывести график, созданный Matplotlib, в виде массива np в оттенках серого с одним каналом. Есть несколько ответов, например этот , который генерирует вывод как RGB, но я не могу найти метод, аналогичный tostring_rgb, для вызова на холсте и получения его в виде одноканального массива в градациях серого.

Answer 1

Вы можете использовать buffer_rgba, чтобы получить базовый буфер, а затем преобразовать его в оттенки серого с помощью вашей любимой формулы, например, отсюда :

import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np

mpl.use('agg')
plt.bar([0,1,2], [1,2,3], color=list('cmy'))
canvas = plt.gcf().canvas
canvas.draw()

img = np.array(canvas.buffer_rgba())
img = np.rint(img[...,:3] @ [0.2126, 0.7152, 0.0722]).astype(np.uint8)

Результат mpl.use('qt5agg'), plt.imshow(img,'gray', vmin=0, vmax=255): введите описание изображения здесь