我试图使用matplotlib的histogram2d绘制一些2D经验概率分布。我希望颜色在几个不同的图上具有相同的比例,但即使我知道结果分布的全局上限和下限,也找不到设置比例的方法。因此,每个色标将从直方图区间的最小高度到最大高度运行,但每个图形的范围都不同。
一个可能的解决方案是强制一个箱子取下限的高度,另一个我的上限。即便如此,这似乎也不是一项非常直接的任务。
答案 0 :(得分:2)
通常,matplotlib中大多数内容的颜色缩放由vmin
和vmax
关键字参数控制。
您必须稍微阅读这些行,但正如文档中提到的那样,hist2d
中的其他kwargs会传递给pcolorfast
。因此,您可以通过vmin
和vmax
kwargs指定颜色限制。
例如:
import numpy as np
import matplotlib.pyplot as plt
small_data = np.random.random((2, 10))
large_data = np.random.random((2, 100))
fig, axes = plt.subplots(ncols=2, figsize=(10, 5), sharex=True, sharey=True)
# For consistency's sake, we'll set the bins to be identical
bins = np.linspace(0, 1, 10)
axes[0].hist2d(*small_data, bins=bins, vmin=0, vmax=5)
axes[1].hist2d(*large_data, bins=bins, vmin=0, vmax=5)
plt.show()