定义二进制matplotlib色彩映射

Question

我尝试使用matplotlib.imshow绘制布尔2d数组，我希望能够选择两种颜色（一种用于true，另一种用于{ {1}}）。似乎这个论点应该作为一个色彩映射传递给false，但在我的情况下，它似乎有点过分，因为我不需要连续，更重要的是，我不知道如何定义一个自定义色图（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)