Numpy array को PIL छवि में बदलने के लिए कैसे करें matplotlib colormap

Question

मुझे एक साधारण समस्या है लेकिन इसका कोई अच्छा समाधान नहीं मिल रहा है।

मैं एक numpy 2 डी सरणी लेना चाहता हूं जो एक ग्रेस्केल छवि का प्रतिनिधित्व करता है, और इसे matplotlib colormaps में से कुछ लागू करते समय इसे एक आरजीबी पीआईएल छवि में परिवर्तित करना चाहता है।

मैं 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 का उपयोग कर जनहित याचिका छवि प्राप्त करते हैं), मुझे आश्चर्य है कि वहाँ है अगर ऐसा करने का एक आसान तरीका नहीं है, क्योंकि यह छवि दृश्यता की एक बहुत ही प्राकृतिक समस्या प्रतीत होता है। मान लें, इस तरह कुछ:

colored_PIL_image = magic_function(array, cmap) 

पढ़ने के लिए धन्यवाद!

Answer 1

काफी व्यस्त एक लाइनर, लेकिन यहाँ यह है:

1. सबसे पहले सुनिश्चित करें कि आपकी numpy सरणी, myarray, 1.0 पर अधिकतम मूल्य के साथ सामान्यीकृत है।
2. कोलोरैप को सीधे myarray पर लागू करें।
3. 0-255 रेंज पर पुनर्विक्रय करें।
4. np.uint8() का उपयोग करके पूर्णांक में कनवर्ट करें।
5. Image.fromarray() का उपयोग करें।

और आपका काम हो गया:

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

plt.savefig() साथ:

im.save() साथ: