使用numba不支持的numpy函数

Question

我目前正在使用一个函数来使用numpy.histogram()和numpy.histogram2d()创建直方图。为了加快使用numba的过程，我尝试使用@jit装饰器在nonpython模式下进行解释，但是numba报告了一个错误，指出numpy.histogram()是不受支持的功能。我的功能看起来像

def make_histograms(X, neurons, bins = 5):
    xy = np.histogram2d(X, neurons, bins)[0]
    x = np.histogram(X, bins)[0]
    y = np.histogram(neurons, bins)[0]

在我的案子上是否还有其他使用numba的方法？任何帮助将非常感激。在此先感谢：）

Answer 1

Numba文档本身提供了一个示例，可以利用Numba来绘制直方图。看起来很方便；但是，实现2d直方图看起来非常抽象和复杂。幸运的是，我发现这个很棒的repo实现了fast_histogram计算，性能提高了甚至超过了Numba。另外，此page比较了他们的表现。 我自己将这个答案留在这里，希望对某人有用。