Scipy kd-tree，每个点都分配了辅助标识符

Question

我正在尝试构建一个具有 3D 坐标的 kd 树，该坐标具有分配给它们的任意整数。例如元组 ([34534.65424, 10957.00023, -79.154323], 32)。

但是，除了 numpy 数组之外，我找不到如何使用 scipy.spatial.cKDTree() 的方法。
标识符甚至可以是 numpy 数组中的第四个字段，但树必须仅基于前三个坐标构建。有人可以阐明这一点吗？目的是将kd-tree节点引用回它所属的段，以便以后比较。

谢谢！

Answer 1

由于树结构和查询的答案是根据索引而不是值给出的，因此您可以简单地将标签保存在一个单独的数组中，顺序相同。

演示：

a = np.random.rand(100, 3)
t = cKDTree(a)
t.tree.indices
# array([13, 12, 18, 19, 20, 47, 54, 55, 83, 88, 82, 40, 31, 10, 45, 22, 15,
#        95, 76, 46, 94, 62,  5, 84, 74, 63, 80, 81, 59,  9, 34, 29, 71, 61,
#        73,  4, 85,  0, 16, 39, 35, 26, 53, 69, 77,  6, 36, 97, 57, 78, 90,
#        32, 38, 68, 65, 96,  3, 24, 66, 11, 86, 89, 91, 30, 67, 42, 28, 33,
#         7, 44,  8, 17, 60, 56, 41, 43, 92, 51, 58,  2, 93, 21, 14, 64, 70,
#        87, 72, 50, 25, 49, 98, 37, 99, 52,  1, 48, 75, 79, 23, 27])

t.query_pairs(0.1)
# {(20, 55), (48, 98), (6, 35), (38, 89), (98, 99), (26, 39), (45, 84), (26, 53), (37, 48), (48, 99), (15, 46), (36, 97), (14, 87), (42, 56), (37, 99), (5, 94), (71, 73), (0, 85), (70, 72), (22, 74), (37, 98)}

Scipy kd-tree，每个点都分配了辅助标识符

Scipy kd-tree with auxiliary identifier assigned to each point

python

numpy

scipy

kdtree