matplotlib利用scatter绘制彩色图像：NameError: name 'array' is not defined

《Machine Learning in Action》，《机器学习实践》

在学习这本书“中文版”的时候，第二章2.2.2中利用matplotlib库函数scatter函数绘制彩色图像

>>> ax.scatter(da[:,1],da[:,2],15.0*array(lab),15.0*array(lab))

Traceback (most recent call last):
  File "", line 1, in 
    ax.scatter(da[:,1],da[:,2],15.0*array(lab),15.0*array(lab))
NameError: name 'array' is not defined

提示错误：NameError: name 'array' is not defined

通过网上查找，发现这是由于没有导入array而导致的。

在这里通过

>>> import kNN
>>> da,lab = kNN.file2matrix('E:\cc\python_练习\datingTestSet2.txt')

>>> import matplotlib
>>> import matplotlib.pyplot as plt
>>> fig = plt.figure()
>>> ax = fig.add_subplot(111)

>>> from numpy import *       #这一步是关键
>>> ax.scatter(da[:,1],da[:,2],15.0*array(lab),15.0*array(lab))

>>> plt.show()

最后显示出来图像：

问题解决。

这里，感谢网友的无私分享，参考了各位的解决方法。谢谢。。。

豆瓣网友： 点击打开链接

stack overflow：  点击打开链接

liuyuehui110的专栏：点击打开链接

豆瓣网友：

python shell 执行代码：

>>> import sys
>>> sys.path.append("D:\python")
>>> import kNN
>>> from numpy import *
>>> import matplotlib
>>> import matplotlib.pyplot as plt
>>> fig = plt.figure()
>>> ax = fig.add_subplot(111)
>>> mat,lab = kNN.file2matrix('D:\python\datingTestSet2.txt')
>>> ax.scatter(mat[:,1], mat[:,2], 15.0*array(map(int,lab)),15.0*array(map(int,lab))) #重点修改本行
>>> plt.show()