Convert 2d array to collored image in python

user_7313914 0
Viewed 4031

I have 2d list of ints like this:

list1 = [[1, 30, 50], [21, 45, 9], [97, 321, 100]]

Next i am going to convert this to numpy array:

myarr = np.asarray(list1)

Next i am going to convert this to Image using PIL like this:

img = Image.fromarray(myarr, "I")
img.save("my.png")

the problem is that i dont want image in grayscale. I dont know how to convert this in to collored image. I have to use eny map function or somethig else ?

python-imaging-library numpy image python
1 Answers

The way to accomplish this is with numpy

import numpy as np
list1 = [[1, 30, 50], [21, 45, 9], [97, 321, 100]]
list1 = np.array(list1).reshape(-1, 3)

And now list1 will have the shape N x 3, where the 3 dimension is RGB. If you know the sizes of the final image, you can do

np.array(list1).reshape(N, M, 3)

and it will re-shape your array into RGB as you need. Then once you have the numpy array, you have your array in the shape of an image and can save it to PNG, etc.

user_57567010
Related