matplotlib hist2d用于空像素的colormap

Question

我发现将空像素设置为白色的一般方法是使用lognorm colormap。

有没有办法使用线性色图，但仍然将空像素设置为白色？

谢谢！

Answer 1

您可以使用'.set_under'：

import matplotlib.pyplot as plt

my_cmap = plt.cm.jet
my_cmap.set_under('w',1)
...
plt.hist2d( ..., cmap = my_cmap)

这将所有低于1的值（即最低计数）设置为'w'（=白色）。

Answer 2

在新版本的matplotlib中，您可以简单地使用cmin = 1，即

plt.hist2d(x,y,cmin = 1)

会将所有计数少于一个的bin设置为nan，即不可见。如果您的背景是白色（默认），则这些垃圾箱将显示为白色。但是它们会改变背景。