How to plot hyperrectangles of scipy.spatial.kdtree

Question

Im trying to perform a KD-tree on a set of x,y coordinate pairs for spatial indexing them. I found scipy.spatial.kdtree to perform this quickly. However I can't seem to manage plotting hyperrectangles of the kdtree (Im using matplotlib).

Does anyone know how to fetch the hyperrectangles generated by the algorithm?

Couldnt attach image, here is my desired output: [A kd-tree.](https://en.wikipedia.org/wiki/File:Kdtree_2d.svg) — Qiicken, Jan 31 '22 at 09:45
Please provide enough code so others can better understand or reproduce the problem. — Community, Feb 09 '22 at 22:36

denis · Answer 1 · 2022-07-29T09:11:42.607

how to fetch the hyperrectangles generated by the algorithm ?

the trick is to generate them as you go along: extend wikipedia K-d tree with e.g.

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
)):
...

Code to build (no query yet) might be under my gists.
A sample plot:

How to plot hyperrectangles of scipy.spatial.kdtree

1 Answers1