Python 使用matplotlib绘制具有可分辨和不连续颜色的颜色栏

下图中的颜色栏是在matlab中绘制的。我想知道是否可以在python/matplotlib中绘制类似的颜色条？我需要在颜色栏中进行取消平滑颜色更改，以便将值0-5表示为非常可分辨和不连续的颜色。如果可能的话，你能给我举个例子吗？谢谢~

使用和：

使用ListedColormap的一些示例：

import numpy as np
import matplotlib as mpl
from matplotlib import pyplot
data = np.floor(np.random.random((10,10)) * 6)
cmap = mpl.colors.ListedColormap(['w', 'b', 'g', 'y', '#ff8c00', 'r'])
norm = mpl.colors.BoundaryNorm([0,1,2,3,4,5,6], cmap.N)
pyplot.imshow(data, cmap=cmap, norm=norm, interpolation='none')
pyplot.colorbar()
pyplot.show()