Python Matplotlib:如何更改标记的颜色?

时间:2018-10-09 14:57:11

标签: python python-3.x matplotlib

我阅读了Packt的《机器学习算法》这本书,并且有一个代码示例,我试图对其进行复制。我在此散点图中的标记颜色有些问题。

代码如下:

from sklearn.datasets import make_circles

nb_samples = 500
X, Y = make_circles(n_samples=nb_samples, noise=0.1)

这将产生一个包含数据的圆。书中的图片如下所示: Picture of the book

我尝试用以下方式重现此内容:

from sklearn.datasets import make_circles
import matplotlib.pyplot as plt

nb_samples = 500
X, Y = make_circles(n_samples=nb_samples, noise=0.1)
plt.scatter(X[:, 0], X[:, 1])
plt.show()

输出如下:

My solution

我想知道如何为数据点设置不同的颜色和标记。也许我的代码是错误的,我不应该绘制X [:, 0],X [:, 1]。我希望有人能帮助我。

1 个答案:

答案 0 :(得分:1)

颜色推测是由make_circles的另一个返回值产生的。因此

plt.scatter(X[:, 0], X[:, 1], c=Y, cmap="bwr")

enter image description here