机器学习实战第九章ValueError: Masked arrays must be 1-D

巧笑倩茜

于 2017-12-19 13:14:34 发布

阅读量4k

点赞数 10

CC 4.0 BY-SA版权

分类专栏： python 机器学习

本文链接：https://blog.youkuaiyun.com/feixiang5527/article/details/78840840

python 同时被 2 个专栏收录

12 篇文章

订阅专栏

机器学习

2 篇文章

订阅专栏

本文解决了《机器学习实战》第九章中的一个常见错误：ValueError: Masked arrays must be 1-D。通过调整绘图函数中的数据处理方式，确保了散点图绘制时数据的一致性和正确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

机器学习实战第九章ValueError: Masked arrays must be 1-D

主要错误出现在函数reDraw(tolS, tolN)中：

def reDraw(tolS, tolN):
    reDraw.f.clf() 
    reDraw.a = reDraw.f.add_subplot(111)
    if chkBtnVar.get():
        if tolN < 2: tolN = 2
        myTree = regTrees.createTree(reDraw.rawDat, regTrees.modelLeaf, \
                                     regTrees.modelErr, (tolS, tolN))
        yHat = regTrees.createForeCast(myTree, reDraw.testDat, \
                                       regTrees.modelTreeEval)
    else:
        myTree = regTrees.createTree(reDraw.rawDat, ops=(tolS, tolN))
        yHat = regTrees.createForeCast(myTree, reDraw.testDat)
    reDraw.a.scatter(reDraw.rawDat[:, 0], reDraw.rawDat[:, 1], s=5)  
    reDraw.a.plot(reDraw.testDat, yHat, linewidth=2.0)
    reDraw.canvas.show()

将其中的

reDraw.a.scatter(reDraw.rawDat[:, 0], reDraw.rawDat[:, 1], s=5)

替换为

reDraw.a.scatter(reDraw.rawDat[:, 0].tolist(), reDraw.rawDat[:, 1].tolist(), s=5)

则运行成功。