Kdtree subset query

@rgommers 
#4260 was merged despite not having any commits inside... So I am reissuing a PR

This should be:

    x : array_like
        Point to query.