Question

我正在创建一个像How to create a density plot in matplotlib?这样的密度图，这样我就可以在地图上找到很少的点，我的问题是我需要知道最高峰的确切x值是多少。

我可以通过循环找到它：

    density = gaussian_kde(data)
    aa = 0
    bb = 0
    for i in range(max value of data):
       if density(i)[0]>aa:
            aa = density(i)[0]
            bb = i

此bb具有峰值的x值但是执行此循环的时间太长。目前大约需要25秒，将来数据的大小会更大

我希望这不是重复但至少我找不到这个问题。

Answer 1

您可以使用numpy.argmax：

ys = density(np.arange(9))
bb = np.argmax(ys)
aa = ys[bb]

这将计算与您发布的代码相同的aa和bb值。但是，这只会在x的整数值中找到最大值。如果你看Justin Peel's graph，你会发现峰值密度可能出现在某个非整数x-value。因此，要找到更接近峰值密度的近似值，请使用

xs = np.linspace(0,8,200)
ys = density(xs)
index = np.argmax(ys)
max_y = ys[index]
max_x = xs[index]

如何在python中获得密度峰值

1 个答案: