matplotlib - matplotlib.colors.LogNorm 中的零问题

Question

我正在使用绘制直方图

plt.imshow(hist2d, norm = LogNorm(), cmap = gray)

其中hist2d是直方图值的矩阵。除了 hist2d 中的元素为零之外，此方法工作正常。特别是，我获得了以下图像

但希望白色斑 block 变成黑色。

谢谢!

Answer 1

这是一种替代方法，不需要您通过为坏像素设置 RGB 值来破坏数据。

import copy
data = np.arange(25).reshape((5,5))
my_cmap = copy.copy(matplotlib.cm.get_cmap('gray')) # copy the default cmap
my_cmap.set_bad((0,0,0))
plt.imshow(data, 
           norm=matplotlib.colors.LogNorm(), 
           interpolation='nearest', 
           cmap=my_cmap)

问题是带有 0 的 bin 无法正确地进行日志规范化，因此它们被标记为“坏”，并且映射到不同的位置。默认行为是不在这些像素上绘制任何内容。您还可以指定绘制超出或低于颜色图限制的像素的颜色(默认是将它们绘制为最高/最低颜色)。