有没有一种方法可以将三值数组转换为图像？
我正在尝试将随机填充数字（0,1,2）的数组转换为图像，其中每个数字都显示为不同的颜色（如果可能的话，最好由我选择），但是我找不到任何图像做到的方式。
I'm trying to convert an array filled randomly with numbers (0,1,2) to an image where every number is shown as a different color (preferably picked by me if possible), but I can't find any way to do it. Is there someone who knows if it can be done?
我尝试使用PIL，但事实证明我的尝试非常不令人满意。如果有人可以提供帮助，我将不胜感激。
I tried to use PIL but my tries are proving to be very unsatisfactory. I'd really appreciate if someone could help.
我知道如何将其显示为图像，但是我没有不知道如何将其随机化。假设我有一个尺寸为400x500的数组，并且我想使每个单元格都具有三个值之一，可以这样做吗？
（这部分代码的大部分来自注释，不是我写的）
I got how to show it as an image, but I don't know how to randomize it. Let's say I have an array with dimensions 400x500 and I'd like to make every cell have one of the three values, can I do something like this? (most of this part of code is from a comment, it's not written by me)
from PIL import Image
import numpy as np
w, h = 500, 400
a = [255, 0, 0]
b = [0, 255, 0]
c = [0, 0, 255]
data = np.array(np.random.random((a,b,c),(h, w, 3), dtype=np.uint8)
#I'd like the random.random to take one of the three values ("[255, 0, 0]", "[0,255, 0]", or "[0, 0, 255]")
img = Image.fromarray(data, 'RGB')
img.save('my.png')
img.show()
有办法吗？
我现在明白了，谢谢大家的帮助！
I got it now, thank you for the help everybody!
您与代码非常接近，只是缺少查找表（LUT）来查找与您的0,1,2数据相对应的颜色：
You were very close with your code, you were just missing a Look Up Table (LUT) to look up the colour corresponding to your 0,1,2 data:
#!/usr/local/bin/python3
import numpy as np
from PIL import Image
# Specify image size, create and fill with random 0, 1, or 2
w, h = 500, 400
data = np.random.randint(0,3,(h, w), dtype=np.uint8)
# Make LUT (Look Up Table) with your 3 colours
LUT = np.zeros((3,3),dtype=np.uint8)
LUT[0]=[255,0,0]
LUT[1]=[0,255,0]
LUT[2]=[0,0,255]
# Look Up each pixel in the LUT
pixels = LUT[data]
# Convert Numpy array to image, save and display
img = Image.fromarray(pixels)
img.save('result.png')
img.show()