如何绘制scipy.spatial.kdtree超矩形

qnyhuwrf  于 2022-11-10  发布在  其他
关注(0)|答案(1)|浏览(111)

我试图在一组x,y坐标对上执行一个KD-tree来对它们进行空间索引。我找到了scipy.spatial.kdtree来快速执行这个操作。但是我似乎无法管理kdtree的超矩形绘制(我使用matplotlib)。
有人知道如何获取算法生成的超矩形吗?

7gcisfzg

7gcisfzg1#

如何获取算法生成的超矩形?
诀窍是在沿着中生成它们:扩展wikipedia K-d tree,例如

class Node( namedtuple( "Node",  # aka Box
"level "    # 0 ..
"axis "     # 0 .. Dim
"mid "      # the point with middle x / middle y
"lo hi "    # corners, inbox(p): lo <= p <= hi
"L R "      # left, right Nodes or None
"points "   # if needed / leaves only
)):
...

要建置的程式码(尚未有查询)可能在my gists之下。
样图:

相关问题