#-*-coding:utf-8-*-
from?numpy?import?*
import?operator
from?os?import?listdir

def?classify0（inX?dataSet?labels?k）:
????dataSetSize?=?dataSet.shape[0]
????#tile（A（mn））???
????#print?dataSet
????#print?“----------------“
????#print?tile（inX?（dataSetSize1））
????#print?“----------------“
????diffMat?=?tile（inX?（dataSetSize1））?-?dataSet??????
????#print?diffMat
????sqDiffMat?=?diffMat**2
????sqDistances?=?sqDiffMat.sum（axis=1）??????????????????
????distances?=?sqDistances**0.5
????sortedDistIndicies?=?distances.argsort（）????????????
????classCount={}??????????????????????????????????????
????for?i?in?range（k）:
????????voteIlabel?=?labels[sortedDistIndicies[i]]
????????classCount[voteIlabel]?=?classCount.get（voteIlabel0）?+?1
????sortedClassCount?=?sorted（classCount.iteritems（）?key=operator.itemgetter（1）?reverse=True）
????return?sortedClassCount[0][0]

def?img2vector（filename）:
????returnVect?=?zeros（（11024））
????fr?=?open（filename）
????for?i?in?range（32）:
????????lineStr?=?fr.readline（）
????????for?j?in?range（32）:
????????????returnVect[032*i+j]?=?int（lineStr[j]）
????return?returnVect

def?handwritingClassTest（）:

????hwLabels?=?[]
????trainingFileList?=?listdir（‘trainingDigits‘）??
????print?trainingFileList????????
????m?=?len（trainingFileList）
????trainingMat?=?zeros（（m1024））
????for?i?in?range（m）:
????????fileNameStr?=?trainingFileList[i]??????????
????????fileStr?=?fileNameStr.split（‘.‘）[0]
????????classNumStr?=?int（fileStr.split（‘_‘）[0]）?
????????hwLabels.append（classNumStr）
????????#print?hwLabels
????????#print?fileNameStr???
????????trainingMat[i:]?=?img2vector（‘trainingDigits/%s‘?%?fileNameStr）
????????#print?trainingMat[i:]?
????????#print?len（trainingMat[i:]）
?????
????testFileList?=?listdir（‘testDigits‘）???????
????errorCount?=?0.0
????mTest?=?len（testFileList）
????for?i?in?range（mTest）:
????????fileNameStr?=?testFileList[i]
????????fileStr?=?fileNameStr.split（‘.‘）[0]?????
????????classNumStr?=?int（fileStr.split（‘_‘）[0]）
????????vectorUnderTest?=?img2vector（‘testDigits/%s‘?%?fileNameStr）
????????classifierResult?=?classify0（vectorUnderTest?trainingMat?hwLabels?3）
????????print?“the?classifier?came?back?with:?%d?the?real?answer?is:?%d“?%?（classifierResult?classNumStr）
????????if?（classifierResult?!=?classNumStr）:?errorCount?+=?1.0
????print?“\nthe?total?number?of?errors?is:?%d“?%?errorCount
????print?“\nthe?total?error?rate?is:?%f“?%?（errorCount/float（mTest））
????
handwritingClassTest（）????

?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????文件????????2464??2015-05-23?21:28??kNN.py
?????文件????????1056??2015-05-10?16:26??testDigits\0_1.txt
?????文件????????1056??2015-05-10?16:26??testDigits\0_2.txt
?????文件????????1056??2015-05-10?16:26??testDigits\0_3.txt
?????文件????????1056??2015-05-10?16:26??testDigits\0_4.txt
?????文件????????1056??2015-05-10?16:26??testDigits\0_5.txt
?????文件????????1056??2015-05-10?16:26??testDigits\1_1.txt
?????文件????????1056??2015-05-10?16:26??testDigits\1_2.txt
?????文件????????1056??2015-05-10?16:26??testDigits\1_3.txt
?????文件????????1056??2015-05-10?16:26??testDigits\1_4.txt
?????文件????????1056??2015-05-10?16:26??testDigits\1_5.txt
?????文件????????1056??2015-05-10?16:26??testDigits\2_1.txt
?????文件????????1056??2015-05-10?16:26??testDigits\2_2.txt
?????文件????????1056??2015-05-10?16:26??testDigits\2_3.txt
?????文件????????1056??2015-05-10?16:26??testDigits\2_4.txt
?????文件????????1056??2015-05-10?16:26??testDigits\2_5.txt
?????文件????????1056??2015-05-10?16:26??testDigits\3_1.txt
?????文件????????1056??2015-05-10?16:26??testDigits\3_2.txt
?????文件????????1056??2015-05-10?16:26??testDigits\3_3.txt
?????文件????????1056??2015-05-10?16:26??testDigits\3_4.txt
?????文件????????1056??2015-05-10?16:26??testDigits\3_5.txt
?????文件????????1056??2015-05-10?16:26??testDigits\4_1.txt
?????文件????????1056??2015-05-10?16:26??testDigits\4_2.txt
?????文件????????1056??2015-05-10?16:26??testDigits\4_3.txt
?????文件????????1056??2015-05-10?16:26??testDigits\4_4.txt
?????文件????????1056??2015-05-10?16:26??testDigits\4_5.txt
?????文件????????1056??2015-05-10?16:26??testDigits\5_1.txt
?????文件????????1056??2015-05-10?16:26??testDigits\5_2.txt
?????文件????????1056??2015-05-10?16:26??testDigits\5_3.txt
?????文件????????1056??2015-05-10?16:26??testDigits\5_4.txt
?????文件????????1056??2015-05-10?16:26??testDigits\5_5.txt
............此處省略120個(gè)文件信息