定义二进制 matplotlib 颜色图

Question

我正在尝试使用matplotlib.imshow来绘制一个布尔二维数组，我希望能够选择两种颜色（一种用于true，一种用于false）。似乎这个参数应该作为颜色图传递给，imshow但在我的情况下它似乎有点矫枉过正，因为我不需要连续，更重要的是，我不知道如何定义自定义颜色图（matplotlib doc没有帮助我）。

Answer 1

import matplotlib.pyplot as plt
import matplotlib.colors

# Color for False and True
cmap = matplotlib.colors.ListedColormap(['red', 'green'])

plt.imshow([True, False], [False, True]], cmap=cmap)