在scipy kdtree中插入对象？

Question

我试图通过scipy的kd_树类创建kd树，该类由对象而不是纯坐标构建。对象有一个（x，y）元组，树就是基于这个元组的，但是我想 将对象本身作为节点/包含在节点中

有什么“简单”的方法吗？ 看了一眼 scipy kdtree with meta data，表示使用第三维作为对象指针（？）。那么，树是否会将此值应用于邻居的比较？ 我也和这位先生在同一条船上，现在跳过创建我自己的kd树就好了

附：这是我的第一篇帖子，所以请对我温柔一点；）

Answer 1

API of scipy's KdTree需要一个二维坐标数组作为输入，而不是任何类型的对象数组。在这个数组中，行是点，列是这些点的坐标

在你链接到的问题中，他不是说有第三维度，而是第三个索引。假设您正在寻找单个最近邻，并且使用某个点query，该函数将返回一个距离和一个索引。索引是对用于构建树的数组的引用。距离是查询点和树点之间的距离差

因此，要使用此树，可以保留两个数组。一个是对象坐标，另一个是对象坐标。它们的顺序应该相同，因此当查询返回索引时，它们在两个数组中的含义相同

这是我的第一个答案，所以也要温柔：D