《机器学习实战》2.2.2 节绘图时出现 NameError：name‘array’is not defined

书中shell如下：

实际运行时 报错 NameError：name‘array’is not defined

修改shell程序如下即可运行：

>>> import kNN  
>>> from numpy import *   
>>> import matplotlib   
>>> import matplotlib.pyplot as plt
>>> fig = plt.figure()
>>> ax = fig.add_subplot(111)
>>> mat,lab = kNN.file2matrix('datingTestSet.txt')
>>> ax.scatter(mat[:,1], mat[:,2], 15.0*array(map(int,lab)),15.0*array(map(int,lab))) 
>>> plt.show()

运行结果如下图：