Как преобразовать массив NumPy в изображение PIL, применяя цветовую карту matplotlib ⇐ Python

[Anonymous]

Я хочу взять 2D-массив NumPy, который представляет изображение в оттенках серого, и преобразовать его в изображение RGB PIL, применяя некоторые цветовые карты matplotlib.
Я могу получить приемлемый PNG вывод с помощью команды pyplot.figure.figimage:
dpi = 100.0
w, h = myarray.shape[1]/dpi, myarray.shape[0]/dpi
fig = plt.figure(figsize=(w,h), dpi=dpi)
fig.figimage(sub, cmap=cm.gist_earth)
plt.savefig('out.png')

Хотя я мог бы адаптировать это, чтобы получить то, что хочу (вероятно, используя StringIO, я получаю образ PIL), мне интересно, нет ли более простого способа сделать это, поскольку кажется, что это так. вполне естественная проблема визуализации изображений. Скажем, примерно так:
colored_PIL_image = magic_function(array, cmap)


Подробнее здесь: https://stackoverflow.com/questions/109 ... b-colormap