定义二进制 matplotlib 颜色图

Question

I'm trying to use the matplotlib.imshow to plot a boolean 2d array and I'd like to be able to pick the two colours (one for true and one for false ). It seems that this argument should be passed as a colormap to imshow but in my case it seems a bit of an overkill as I don't need continuous and, more important, I don't know how to define a custom colormap (the matplotlib doc hasn't helped me with that).

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)