在 hist2d (matplotlib) 中的每个 bin 中打印值

Question

我想创建一个二维直方图，在每个 bin 中，该 bin 表示的值显示在给定 bin 的中心。例如，大小为 5x5 的 hist2d 在最终图中将有 25 个值。这对 PyROOT 来说是可行的，但我需要在这里使用 matplotlib/pyplot。

已根据第一个答案尝试了以下方法：

fig, ax = plt.subplots()
ax.set_aspect("equal")
hist, xbins, ybins, im = ax.hist2d(x, y, bins=(4, [1,2,3,5,10,20]))
ax.text(xbins[1]+0.5,ybins[1]+0.5, "HA", color="w", ha="center", va="center", fontweight="bold")

img = StringIO.StringIO()
plt.savefig(img, format='svg')
img.seek(0)
print("%html <div style='width:500px'>" + img.getvalue() + "</div>")

没有任何错误消息，但“HA”根本不会显示在第一个 bin 中。我在 Zeppelin 中对此进行编程，因此我需要从缓冲区中获取 img ...

Answer 1

要注释 hist2d 绘图，就像任何其他绘图一样，您可以使用 matplotlib 的 text 方法。要注释的值由返回的直方图给出。注释的位置由直方图边缘（加上一半的 bin 宽度）给出。然后，您可以遍历所有 bin 并在每个 bin 中放置一个文本。

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(1)

x = np.random.poisson(size=(160))
y = np.random.poisson(size=(160))

fig, ax = plt.subplots()
ax.set_aspect("equal")
hist, xbins, ybins, im = ax.hist2d(x,y, bins=range(6))

for i in range(len(ybins)-1):
    for j in range(len(xbins)-1):
        ax.text(xbins[j]+0.5,ybins[i]+0.5, hist.T[i,j], 
                color="w", ha="center", va="center", fontweight="bold")

plt.show()

如果只需要一个注释，例如以下

ax.text(xbins[1]+0.5,ybins[1]+0.5, "HA", 
        color="w", ha="center", va="center", fontweight="bold")

会产生