如何为图像中的每个像素分配特定颜色

如何解决如何为图像中的每个像素分配特定颜色

我有一个带有 5 个值的单波段光栅 tif 图像，即 4,3,2,1,0 。

此代码显示图像，随机颜色分配给每个像素值

import Rasterio
from Rasterio.plot import show
import matplotlib.pyplot as plt

img_data = Rasterio.open(img)
fig,ax = plt.subplots(figsize=(10,10))    
show(img_data)

如何通过为每个图像分配特定颜色来显示此图像（例如红色：4，蓝色：3，绿色：2，黑色：1，白色：0）。 我遇到了颜色图选项，但无法按需要显示它。