python - 配列内の数値に関連付けられたカラーバー

Question

0 から 3 までの 155 個の数字の配列があります。各数字を異なる色に対応させたいと考えています。したがって、赤 0、青 1、緑 2、黄 3 で、配列をカラーバーにプロットします。

origin = 'lower'
#origin = 'upper'

delta = 0.025

x = y = np.arange(-3.0, 3.01, delta)
X, Y = np.meshgrid(x, y)
Z1 = plt.mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
Z2 = plt.mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
Z = 10 * (Z1 - Z2)
levels = salist ##array of 155
CS3 = plt.contourf(X, Y, Z, levels,
                    colors = ('r', 'g', 'b','y'),
                    origin=origin,
                    extend='both')

plt.colorbar(CS3)