小编lth*_*ner的帖子

我想使用 matplotlib 颜色图的 set_bad 参数为 nan 值分配特定颜色。我创建了一个颜色图：

from matplotlib.colors import ListedColormap, BoundaryNorm

cmap = ListedColormap(['green', 'yellow', 'red'])
norm = BoundaryNorm([0, 20, 70, 100], cmap.N)

然后我将坏颜色设置为黑色：

cmap.set_bad(color="k")

为了验证一切按预期工作，我绘制了一个数组作为测试，并且它有效（左下象限是黑色）：

import numpy as np
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
im = ax.imshow(np.array([[10, 50], [np.nan, 70]]), cmap=cmap, norm=norm)
fig.colorbar(im, ax=ax)

但是，如果我在集合上使用相同的颜色图，则它不起作用：

from matplotlib.collections import LineCollection
lc = LineCollection(([(0.1, 0.1), (0.1, 0.2)], [(0.2, .2), (0.2, .3)], [(0.3, .3), (0.3, .4)]))

lc.set_cmap(cmap)
lc.set_norm(norm)
lc.set_array(np.array([10, np.nan, 80]))

fig, ax = plt.subplots()
ax.add_collection(lc)
plt.draw()

中间的线有一个 …