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

2024-03-28 14:44:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我读了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]。我希望有人能帮助我。你知道吗


Tags: 数据代码from标记importmake颜色plt
1条回答
网友
1楼 · 发布于 2024-03-28 14:44:51

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

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

enter image description here

相关问题 更多 >