定义二进制 matplotlib 颜色图

Question

我正在尝试使用

matplotlib.imshow

绘制布尔二维数组，我希望能够选择两种颜色（一种用于

true

，一种用于

false

）。似乎这个参数应该作为颜色图传递给

imshow

但就我而言，这似乎有点矫枉过正，因为我不需要连续，更重要的是，我不知道如何定义自定义颜色图（ matplotlib 文档没有帮助我）。

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)
plt.show()