python-2.7 - 有什么方法可以在不绘制直方图的情况下使用 matplotlib.pyplot 创建直方图？

Question

我正在使用 matplotlib.pyplot 创建直方图。我实际上对这些直方图的绘图不感兴趣，但对频率和 bin 感兴趣(我知道我可以编写自己的代码来做到这一点，但更喜欢使用这个包)。

我知道我可以做到以下几点

import numpy as np
import matplotlib.pyplot as plt

x1 = np.random.normal(1.5,1.0)
x2 = np.random.normal(0,1.0)

freq, bins, patches = plt.hist([x1,x1],50,histtype='step')

创建直方图。我只需要 freq[0] 、 freq[1] 和 bins[0] 。当我尝试使用时出现问题，

freq, bins, patches = plt.hist([x1,x1],50,histtype='step')

在一个函数中。例如，

def func(x, y, Nbins):
    freq, bins, patches = plt.hist([x,y],Nbins,histtype='step') # create histogram

    bincenters = 0.5*(bins[1:] + bins[:-1]) # center bins

    xf= [float(i) for i in freq[0]] # convert integers to float
    xf = [float(i) for i in freq[1]]

    p = [ (bincenters[j], (1.0 / (xf[j] + yf[j] )) for j in range(Nbins) if (xf[j] + yf[j]) != 0]

    Xt = [j for i,j in p] # separate pairs formed in p
    Yt = [i for i,j in p]

    Y = np.array(Yt) # convert to arrays for later fitting
    X = np.array(Xt)

    return X, Y # return arrays X and Y

当我调用 func(x1,x2,Nbins) 并绘制或打印 X 和 Y 时，我没有得到预期的曲线/值。我怀疑这与 plt.hist 有关系，因为我的图中有部分直方图。

Answer 1

我不知道我是否很好地理解了您的问题，但是在这里，您有一个非常简单的自制直方图示例(一维或二维)，每个直方图都在一个函数中，并正确调用:

import numpy as np
import matplotlib.pyplot as plt

def func2d(x, y, nbins):
    histo, xedges, yedges = np.histogram2d(x,y,nbins)
    plt.plot(x,y,'wo',alpha=0.3)
    plt.imshow(histo.T, 
               extent=[xedges.min(),xedges.max(),yedges.min(),yedges.max()],
               origin='lower', 
               interpolation='nearest', 
               cmap=plt.cm.hot)
    plt.show()

def func1d(x, nbins):
    histo, bin_edges = np.histogram(x,nbins)
    bin_center = 0.5*(bin_edges[1:] + bin_edges[:-1])
    plt.step(bin_center,histo,where='mid')
    plt.show()

x = np.random.normal(1.5,1.0, (1000,1000))

func1d(x[0],40)
func2d(x[0],x[1],40)

当然，您可以检查数据的居中是否正确，但我认为该示例显示了有关此主题的一些有用信息。

我的建议:尽量避免代码中出现任何循环!他们扼杀了表演。如果你看，在我的例子中没有循环。 python数值问题的最佳实践是避免循环! Numpy 有很多 C 实现的函数来完成所有的硬循环工作。