使用Matplotlib以圖例樣式繪制離散顏色條

Question

有時，我想以pcolormesh樣式繪制離散值。

例如，代表100x100形狀的二維數組，其中包含int 0〜7

data  = np.random.randint(8, size=(100,100))
cmap = plt.cm.get_cmap('PiYG', 8) 
plt.pcolormesh(data,cmap = cmap,alpha = 0.75)
plt.colorbar()  

該圖顯示如下：

如何以圖例樣式生成顏色條。 換句話說，每個顏色框對應其值（例如，粉紅色的顏色框-> 0）

這里有一個例子（不適合這個例子）：

Answer 1

也許最簡單的方法是創建相應數量的Patch實例：

import matplotlib.patches as mpatches
import matplotlib.pyplot as plt
import numpy as np

data  = np.random.randint(8, size=(100,100))
cmap = plt.cm.get_cmap('PiYG', 8) 
plt.pcolormesh(data,cmap = cmap,alpha = 0.75)
# Set borders in the interval [0, 1]
bound = np.linspace(0, 1, 9)
# Preparing borders for the legend
bound_prep = np.round(bound * 7, 2)
# Creating 8 Patch instances
plt.legend([mpatches.Patch(color=cmap(b)) for b in bound[:-1]],
           ['{} - {}'.format(bound_prep[i], bound_prep[i+1] - 0.01) for i in range(8)])