From: Milton Cezar Ribeiro
I have an object that is a matrix (i.e. a list of lists)
and I neet do plot it on screen as an image.
My image have values 1 and 2, and I would
like set different colors for each level.
I tryed figure it out without success. In fact
my images are 512x512, but I include a 10x10 sample below.
mymatrix=[[1,2,2,2,1,2,2,2,1,1],
[2,2,1,1,1,1,2,2,1,2],
....
It's best to use a palettized image, since your data points are integers.
First, create one of the right size:
from PIL import Image
im = Image.new('P', (512,512)) # 'P' for palettized
Then, set the image data. You must use a flat list, so you need to flatten
your matrix:
data = sum(matrix, []) # flatten data
im.putdata(data)
You also have to set the palette colors for the used indices (1 and 2). The
palette is a list of 768 ints (256*3), but it's somewhat easier to start
with a list of 3-tuples:
pal = [(0,0,0) for i in range(256)] # all black
pal[1] = (255,0,0) # juicy red
pal[2] = (0,255,0) # crazy green
pal = sum(pal, ()) # flatten it
im.setpalette(pal)
That's it. You should be able to test the result with this:
im.show()
_______________________________________________
Image-SIG maillist - Image-SIG@python.org
http://mail.python.org/mailman/listinfo/image-sig
