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?

Please provide enough code so others can better understand or reproduce the problem. — Community
– Community Bot, Commented Feb 9, 2022 at 22:36

denis · Accepted Answer · 2022-07-29 09:11:42Z

0

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:

edited Jul 29, 2022 at 9:11

answered Jul 28, 2022 at 16:50

denis

22k12 gold badges68 silver badges92 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to plot hyperrectangles of scipy.spatial.kdtree

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related