将matplotlib直方图除以最大仓位值

Question

我想在同一个图上绘制多个直方图，我需要比较数据的分布。我想用每个直方图除以它的最大值，这样所有分布都有相同的比例。然而，根据matplotlib的直方图函数的工作方式，我还没有找到一种简单的方法来实现这一点。

这是因为n在

n, bins, patches = ax1.hist(y, bins = 20, histtype = 'step', color = 'k')

是计数的数量在每个箱，但我不能重传这到hist，因为它将重新计算。

我尝试了范数和密度函数，但这些函数使分布的面积正常化，而不是分布的高度。我可以复制n，然后使用箱输出重复箱边，但这很乏味。当然，hist函数必须允许bins值除以常量？

下面是示例代码，演示了这个问题。

y1 = np.random.randn(100)
y2 = 2*np.random.randn(50)
x1 = np.linspace(1,101,100)
x2 = np.linspace(1,51,50)
gs = plt.GridSpec(1,2, wspace = 0, width_ratios = [3,1])
ax = plt.subplot(gs[0])
ax1 = plt.subplot(gs[1])
ax1.yaxis.set_ticklabels([])   # remove the major ticks

ax.scatter(x1, y1, marker='+',color = 'k')#, c=SNR, cmap=plt.cm.Greys)
ax.scatter(x2, y2, marker='o',color = 'k')#, c=SNR, cmap=plt.cm.Greys)
n1, bins1, patches1 = ax1.hist(y1, bins = 20, histtype = 'step', color = 'k',linewidth = 2, orientation = 'horizontal')
n2, bins2, patched2 = ax1.hist(y2, bins = 20, histtype = 'step', linestyle = 'dashed', color = 'k', orientation = 'horizontal')

Answer 1

为比较而设置的略有不同的方法。可以适应步骤样式：

# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np

y = []
y.append(np.random.normal(2, 2, size=40))
y.append(np.random.normal(3, 1.5, size=40))
y.append(np.random.normal(4,4,size=40))
ls = ['dashed','dotted','solid']

fig, (ax1, ax2, ax3) = plt.subplots(ncols=3)
for l, data in zip(ls, y):
    n, b, p = ax1.hist(data, normed=False,
                       #histtype='step', #step's too much of a pain to get the bins
                       #color='k', linestyle=l,
                       alpha=0.2
                       )
    ax2.hist(data, normed=True,
             #histtype = 'step', color='k', linestyle=l,
             alpha=0.2
             )

    n, b, p = ax3.hist(data, normed=False,
                       #histtype='step', #step's too much of a pain to get the bins
                       #color='k', linestyle=l,
                       alpha=0.2
                       )
    high = float(max([r.get_height() for r in p]))
    for r in p:
        r.set_height(r.get_height()/high)
        ax3.add_patch(r)
    ax3.set_ylim(0,1)

ax1.set_title('hist')
ax2.set_title('area==1')
ax3.set_title('fix height')
plt.show()

一对输出：

Answer 2

您可以将参数bin指定为一个值列表。使用np。arange（）或np。linspace（）生成值。http://matplotlib.org/api/axes_api.html?highlight=hist#matplotlib.axes.Axes.hist

Answer 3

我不知道matplotlib是否默认允许这种正常化，但我自己写了一个函数来实现。

它从plt获取n和bin的输出。hist（如上所述），然后通过下面的函数传递此信息。

def hist_norm_height(n,bins,const):
    ''' Function to normalise bin height by a constant. 
        Needs n and bins from np.histogram or ax.hist.'''

    n = np.repeat(n,2)
    n = float32(n) / const
    new_bins = [bins[0]]
    new_bins.extend(np.repeat(bins[1:],2))
    return n,new_bins[:-1]

现在绘制（我喜欢步骤直方图），将其传递给plt.step.

例如plt。步骤（新的容器，n）。这将为您提供一个直方图，其中高度由一个常量标准化。