Làm thế nào để chuyển đổi mảng Numpy sang hình ảnh PIL áp dụng matplotlib colormap

Question

Tôi có một vấn đề đơn giản nhưng không thể tìm thấy một giải pháp tốt cho nó.

Tôi muốn lấy một mảng 2D dạng khối đại diện cho hình ảnh thang độ xám và chuyển đổi hình ảnh thành hình ảnh RGB PIL trong khi áp dụng một số bản đồ màu matplotlib.

tôi có thể có được một sản lượng PNG hợp lý bằng công pyplot.figure.figimage lệnh:

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

Mặc dù tôi có thể thích ứng này để có được những gì tôi muốn (có thể sử dụng StringIO làm có được hình ảnh PIL), tôi tự hỏi nếu có không phải là một cách đơn giản hơn để làm điều đó, vì nó có vẻ là một vấn đề rất tự nhiên về trực quan hóa hình ảnh. Giả sử như sau:

colored_PIL_image = magic_function(array, cmap) 

Cảm ơn bạn đã đọc!

Answer 1

Khá bận rộn lót, nhưng ở đây nó là:

1. Đầu tiên đảm bảo mảng NumPy của bạn, myarray, là bình thường với giá trị tối đa tại 1.0.
2. Áp dụng bản đồ màu trực tiếp lên myarray.
3. Rescale đến phạm vi 0-255.
4. Chuyển đổi thành số nguyên, sử dụng np.uint8().
5. Sử dụng Image.fromarray().

Và bạn đã hoàn tất:

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

với plt.savefig():

với im.save():