Matplot Scatter График разных цветов на основе значения столбца данных кадра - PullRequest
0 голосов
/ 23 февраля 2020

Может быть, мой вопрос уже был дан ответ здесь, но я вроде не нашел решение, работающее на меня. Что мне нужно сделать: я хочу создать график рассеяния из столбца кадра данных, и я хочу разные цвета для точек рассеяния в зависимости от значения. В столбце есть только три значения 0,1 и 2.

Я попробовал это так:

plt.scatter(train.index, train['target'] == 0, c = '#009682', marker = 'o')
plt.scatter(train.index, train['target'] == 1, c = '#808080', marker = ',')
plt.scatter(train.index, train['target'] == 2, c = '#4664aa', marker = '+')

train - это фрейм данных, который я использую. Я также пытался использовать заговор вместо разброса, но он тоже не работал. заранее спасибо:)

...