带标准化行的二维直方图

Question

我想画一个二维柱状图，每一行标准化。我正在使用以下代码：

H, xedges, yedges = np.histogram2d(charged_zgs, zgs, weights=prescales, normed=1, range=[[0, 0.5], [0, 0.5]], bins=25)
Hmasked = np.ma.masked_where(H==0,H) # Mask pixels with a value of zero

plt.pcolormesh(xedges,yedges,Hmasked)

cbar = plt.colorbar()
cbar.ax.set_ylabel('Counts')  

plt.show()

但是，normed=1位使二重积分为1，而我希望每个单独的行都规范化。在

我想我可以使用一个元组列表来表示权重，我可以手动规范化这些权重，但后来意识到参数权重必须是1D列表。在

Answer 1

您可以使用keepdims=True来避免所有的转置：

H_norm_rows = H / H.max(axis=1, keepdims=True)

Answer 2

如果要规范化每一行，可以使用：

H_norm_rows = (H.T/np.sum(H,axis=1)).T