使用自定义颜色显示阵列

Question

我有一个名为myArray的2D numpy数组，由1到5的整数组成。 我想使用matplotlib显示这个数组，确保值1显示为绿色，2显示为黄色，3显示为蓝色，4显示为黑色，5显示为红色

我写了以下几行，但颜色与值不对应，因为我不知道如何处理这些

import matplotlib.pyplot as plt
from matplotlib import colors
myColors = ["green", "yellow", "blue", "black", "red"]
myCmap = colors.ListedColormap(myColors)
plt.imshow(myArray, cmap = myCmap)