построение изображения из прогнозируемого значения

Question

У меня есть прогнозируемое значение 'p', и когда я печатаю 'p', получается:

[[ 6.8175583]
[ 9.99688  ]
[12.834187 ]
...
[ 1.4792776]
[ 3.741147 ]
[ 8.88671  ]]

Есть ли способ, которым я могу построить это как изображение?

когда Я использую plt.imshow(p), он показывает мне одну строку:

Answer 1

Вы можете преобразовать массив NumPy в изображение PIL, используя режим Image.fromarray:

import numpy as np
from PIL import Image 

#assuming p is your array 
#w, h are the width and height you want for your image
img = Image.fromarray(p.reshape(w,h), 'L')
img.save('my.png')
img.show()

L, указывающий, что значения массива представляют яркость. Результатом будет серое изображение.