灰度到三色转换

Question

I would like to change the output PNG of this script to tri-color (red, blue, yellow)

Instead of its current grayscale format.

I figure I need to change lines 34-45, but im not sure how to go about it.

img = Image.new("L", size)  # grayscale, blank black image

ind = 0

for row in range(0, size[0]):

    for col in range(0, size[1]):

        if ind < text_length:  # only change pixel value for length of text

            pixel_value = convert_char_to_int(text[ind], limit=limit)

            img.putpixel((row, col), pixel_value)

            ind += 1

        else:  # end of text, leave remaining pixel(s) black to indicate null

            break

img.save(result_path)

return result_path

Any advice would be useful!

Answer 1

According to the PIL documentation for Image.putpixel() :

im.putpixel(xy, colour)

Modifies the pixel at the given position. The colour is given as a single numerical value for single-band images, and a tuple for multi-band images.

So, to insert a coloured pixel using RGB format:

img = Image.new("RBG", size)  # RGB image

img.putpixel((row, col), (r,g,b))

You may also be interested in Image.convert()