matplotlib：二进制热图

Question

假设我有一个10x10的矩阵，由10或1组成，表示为列表列表。我怎么能用matplotlib红色和黑色正方形的网格表示这种矩阵？（红色代表1，黑色代表0）。

我已经进行了广泛的搜索，但是我能找到的最接近的是Plot a black-and-white binary map in matplotlib，但是颜色的浮点数小于1。一旦我在数据中得到1，该图就会出错。有人可以帮忙吗？还是指向特定的matplotlib文档来帮助我克服这一问题？

Answer 1

您需要所谓的ListedColorMap：

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

# random data
x = np.random.random_integers(0, 1, (10, 10))

fig, ax = plt.subplots()

# define the colors
cmap = mpl.colors.ListedColormap(['r', 'k'])

# create a normalize object the describes the limits of
# each color
bounds = [0., 0.5, 1.]
norm = mpl.colors.BoundaryNorm(bounds, cmap.N)

# plot it
ax.imshow(x, interpolation='none', cmap=cmap, norm=norm)