Python Kd Tree Query Border