Python Matplotlib: Как изменить цвет маркеров? - PullRequest
0 голосов
/ 09 октября 2018

Я прочитал книгу «Алгоритмы машинного обучения» от 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 голосов
/ 09 октября 2018

Цвет предположительно создается другим возвращаемым значением make_circles.Следовательно

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

enter image description here

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...