OpenCV with Python - KNN Example

KNN Ref:
[url]http://www.hudong.com/wiki/KNN[/url]
[url]http://en.wikipedia.org/wiki/K-nearest_neighbor_algorithm[/url]

Example Ref: (C Language)
[url]http://www.opencv.org.cn/index.php/%E6%9C%BA%E5%99%A8%E5%AD%A6%E4%B9%A0%E4%B8%AD%E6%96%87%E5%8F%82%E8%80%83%E6%89%8B%E5%86%8C[/url]

from ml import *
from highgui import *
from cv import *

if __name__ == '__main__':
    color = CV_RGB(180, 120, 0)
    K = 10
    trainSampleCount = 300
    rngState = cvRNG()
    trainData = cvCreateMat(trainSampleCount, 2, CV_32FC1)
    trainClasses = cvCreateMat(trainSampleCount, 1, CV_32FC1)
    img = cvCreateImage(cvSize(500, 500), 8, 3)
    sample = cvCreateMat(1, 2, CV_32FC1)
    cvZero(img)

    # form the training samples
    trainData1 = cvGetRows(trainData, 0, 100)
    colDat