首先,我们创建名为kNN.py的python模块。
from numpy import *
import operator
def createDataSet():
group=array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]])
labels=['A','A','B','B']
return group,labels
我们直接引用这个模块是会报错的。
In [15]: import kNN
Traceback (most recent call last):
File "<ipython-input-15-45c55ea88398>", line 1, in <module>
import kNN
ImportError: No module named kNN
要在python shell中调用这个函数,进入python交互开发环境
我们先使用os模块,
**查看当前路径 os.getcwd()
更改当前路径 os.chdir()**
In [17]: import os
In [19]: os.getcwd()
Out[19]: 'C:\\Users\\Administrator'
我们将kNN.py模块存放在 ‘C:\Users\Administrator’ 的路径下面。
In [20]: import kNN
In [22]: group,labels=kNN.createDataSet()
In [23]: group
Out[23]:
array([[ 1. , 1.1],
[ 1. , 1. ],
[ 0. , 0. ],
[ 0. , 0.1]])
In [24]: labels
Out[24]: ['A', 'A', 'B', 'B']
这样我们发现可以进入kNN模块,并且验证了定义的函数。