ENH: spatial: add query_subset to KDTree (#4236)
Fixes py3 and numpy compatability issues