python KDtree

python KDtree

#!/usr/bin/env python
# coding: utf-8
import numpy as np

trainSet = [[2,3],[5,4],[9,6],[4,7],[8,1],[7,2]]
trainSet1 = [[2,3,1],[5,4,3],[9,6,7],[4,7,3],[8,1,4],[7,2,6]]

class kdtree:
    """kdtree node definition"""    
    class node():
        def __init__(self, point, left