python - 概率网格 matplotlib

Question

我想获取将值映射到颜色的网格。它应该像一个概率分布，其中 1.0 对应于黑色，0.0 对应于白色，灰色介于两者之间。阅读答案 here我尝试一下我的案例:

from matplotlib import mpl,pyplot
import numpy as np

zvals = np.zeros((10,5))
zvals[5,1] = 0.5
zvals[5,2] = 0.3
zvals[5,3] = 0.7
zvals[5,4] = 5.8

cmap = mpl.colors.LinearSegmentedColormap.from_list('my_colormap',['white','black'],256)

img = pyplot.imshow(zvals,interpolation='nearest', cmap = cmap)

pyplot.colorbar(img,cmap=cmap)
pyplot.show()

我看到了这个问题，因为线性映射将 2D 数组中的最大值视为黑色。但数组不一定包含值 1.0。

如何设置 0.0 和 1.0 的硬边界，并在其间平滑过渡？

Answer 1

使用 imshow 的 vmin 和 vmax 参数:

img = pyplot.imshow(zvals, interpolation='nearest', cmap=cmap, vmin=0, vmax=1)