Python Matplotlib：如何为图例添加“不良”颜色？

Question

matplotlib中的颜色图允许使用特定颜色标记“不良”值，即NaN。当我们随后绘制颜色条时，不包括该颜色。是否有一种首选的方法可以同时显示连续的颜色条和不连续的图例，以显示特定颜色的不良值？]

代码示例：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm

data = np.random.rand(10, 10)
data[0:3, 0:3] = np.nan  # some bad values for set_bad

colMap = cm.RdBu
colMap.set_bad(color='black')

plt.figure(figsize=(10, 9))
confusion_matrix = plt.imshow(data, cmap=colMap, vmin=0, vmax=1)
plt.colorbar(confusion_matrix)
plt.show()

哪个生产：

Answer 1

您可以使用https://matplotlib.org/3.1.1/tutorials/colors/colorbar_only.html#discrete-intervals-colorbar所示的超出范围绘图的一种方法来执行此操作

[设置错误值的颜色，例如到-999并使用关键字extend。

另一种方法是使用如图here所示的蒙版图。