Python - 获取最密集点的坐标

时间:2014-10-05 23:35:19

标签: python numpy scipy kernel-density argmax

我正在使用numpy和scipy从3D坐标信息生成密度图。我可以通过使用以下代码生成KDE来成功生成数据的密度图

xyz = np.vstack([x,y,z])
kde = stats.gaussian_kde(xyz)
density = kde(xyz)

但是,如何使用此信息查找与最大密度的3D点相关联的坐标?

我试过

max(density)

返回一个值,然后我可以找到

的索引
density.argmax(axis=0)

然后我打了一个空白,因为我似乎无法使用该索引从xyz中获取相关坐标,我不确定这是否是正确的方法。

1 个答案:

答案 0 :(得分:2)

here,我可以使用

xyz.T[np.argmax(density)]

返回数据中最密集点的3D坐标