Cómo convertir matriz Numpy a imagen PIL aplicando matplotlib colormap

Question

Tengo un problema simple pero no puedo encontrar una buena solución para él.

Quiero tomar una matriz numpy 2D que representa una imagen en escala de grises, y convertirla en una imagen PIL RGB mientras aplica algunos de los mapas de color matplotlib.

puedo obtener una salida PNG razonable utilizando el comando 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') 

Aunque podría adaptar esta a conseguir lo que quiero (probablemente utilizando StringIO hacer llegar la imagen PIL), me pregunto si hay no es una forma más sencilla de hacerlo, ya que parece ser un problema muy natural de visualización de imágenes. Digamos algo como esto:

colored_PIL_image = magic_function(array, cmap) 

¡Gracias por su lectura!

Answer 1

Bastante ocupado un trazador de líneas, pero aquí está:

1. En primer lugar asegurarse de que su matriz numpy, myarray, se normaliza con el valor máximo en 1.0.
2. Aplicar el mapa de colores directamente al myarray.
3. Escala al rango 0-255.
4. Convierte a enteros, usando np.uint8().
5. Utilice Image.fromarray().

y ya está:

from PIL import Image 
im = Image.fromarray(np.uint8(cm.gist_earth(myarray)*255)) 

con plt.savefig():

con im.save():