matplot散点图根据数据框的列值绘制不同的颜色

时间:2020-02-23 14:13:12

标签: python dataframe matplotlib plot scatter

也许我的问题已经在这里回答了,但我还没有找到适合我的解决方案。 我需要做的是:我想从数据框的列创建散点图,并且我希望基于该值为散点指定不同的颜色。列中只有三个值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是我正在使用的数据框。我也尝试使用图来代替散点图,但是它也不起作用。在此先感谢:)

0 个答案:

没有答案