BOBO老师，执行同样的语句，报错

Question

plt.scatter(X[y0,0],X[y0:,1],color=“red”)
plt.scatter(X[y1,0],X[y1:,1],color=“green”)
plt.scatter(X[y2,0],X[y2:,1],color=“blue”)

---

TypeError Traceback (most recent call last)
in ()
----> 1 plt.scatter(X[y0,0],X[y0:,1],color=“red”)
2 plt.scatter(X[y1,0],X[y1:,1],color=“green”)
3 plt.scatter(X[y2,0],X[y2:,1],color=“blue”)

TypeError: only integer scalar arrays can be converted to a scalar index

y==0
array([ True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, True, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False])
是不是因为类型问题？数组的内容不是int数据，应该怎么转换呢？

解决了：我多了个冒号，X[y0:,1]

Answer 1

liuyubobobo 2018-11-11 13:51:04

你的代码和课程中不一样。课程中的写法是：

plt.scatter(X[y==0,0], X[y==0,1], color="red")
plt.scatter(X[y==1,0], X[y==1,1], color="blue")
plt.scatter(X[y==2,0], X[y==2,1], color="green")
plt.show()

为什么使用 y == 0 这样的方式，可以参考这个问答：

https://coding.imooc.com/learn/questiondetail/36343.html

加油！：）

• 提问者 qq_流光长河_0 #1

  非常感谢老师！我刚发现。

• 提问者 qq_流光长河_0 #2

  y==0 跟 np.argwhere(y==0) 的意思一样，对吧？

• 提问者 qq_流光长河_0 #3

  感觉不一样，但是执行效果一样

点开查看后面1条评论