Come convertire l'array Numpy in immagine PIL applicando la mappa colori matplotlib

Question

Ho un problema semplice ma non riesco a trovare una buona soluzione.

Voglio prendere una matrice 2D numpy che rappresenta un'immagine in scala di grigi e convertirla in un'immagine RGB PIL mentre si applicano alcune delle mappe dei colori Matplotlib.

posso ottenere un'uscita PNG ragionevole utilizzando il 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') 

Anche se ho potuto adattare questo per ottenere ciò che voglio (probabilmente usando StringIO si ottiene l'immagine PIL), mi chiedo se c'è non è un modo più semplice per farlo, dal momento che sembra essere un problema molto naturale di visualizzazione delle immagini. Diciamo, qualcosa del genere:

colored_PIL_image = magic_function(array, cmap) 

Grazie per la lettura!

Answer 1

Piuttosto un occupato uno di linea, ma qui è:

1. Primo garantire l'array NumPy, myarray, è normalizzata con il valore massimo a 1.0.
2. Applica la mappa dei colori direttamente a myarray.
3. Rescale fino alla gamma 0-255.
4. Convertire in numeri interi, utilizzando np.uint8().
5. Utilizzare Image.fromarray().

E il gioco è fatto:

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

con plt.savefig():

con im.save():