Numpy dizisi, PIL görüntüsüne dönüştürme matplotlib colormap uygulamasına dönüştürülüyor

Question

Basit bir sorunum var ama buna iyi bir çözüm bulamıyorum.

Gri ​​tonlamalı bir görüntüyü temsil eden ve matplotlib renk haritalarından bazılarını uygularken bir RGB PIL görüntüsüne dönüştüren bir sayısal 2D dizi almak istiyorum.

Ben pyplot.figure.figimage komutunu kullanarak makul PNG çıktı alabilirsiniz : yoktur İstediğimi elde etmek için bu uyum olabilir rağmen

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') 

(muhtemelen StringIO kullanarak PIL görüntü elde yapmak), ben acaba eğer Bunu yapmak için daha basit bir yol değil, çünkü görüntü görselleştirmesinin çok doğal bir sorunu gibi görünüyor. Diyelim ki şöyle bir şey:

colored_PIL_image = magic_function(array, cmap) 

Okumak için teşekkürler!

Answer 1

Oldukça yoğun bir liner, ama işte burada:

1. Önce numpy dizi myarray, 1.0 de maksimum değere sahip normalize olmasını sağlar.
2. Renk eşlemesini doğrudan myarray'a uygulayın.
3. 0-255 ürün yelpazesine yeniden boyutlandırın.
4. Tam sayılara dönüştür, np.uint8()'u kullanarak.
5. Image.fromarray()'u kullanın.

Ve bitirdiniz:

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

plt.savefig() ile:

im.save() ile: