如何在matplotlib中创建线图,并根据类别变量使用不同的颜色?

时间:2018-08-08 17:25:28

标签: python-3.x matplotlib charts colors

我有一个折线图要绘制。我希望此折线图沿线是不同的颜色,具体取决于分类变量。

我想这与部署色调等效,尽管在matplotlib中部署线图时似乎并不那么简单。

作为使用示例:

import numpy as np
import pandas as pd
df = pd.DataFrame({'X':[1,2,3,4,5,3,2,6,8,10],'Category':['Apple','Apple','Apple','Blueberry','Blueberry','Blueberry','Blueberry','Raspberry','Raspberry','Raspberry']},index=[np.arange(1,11)])
df

这将提供以下df:

    Category    X
1   Apple   1
2   Apple   2
3   Apple   3
4   Blueberry   4
5   Blueberry   5
6   Blueberry   3
7   Blueberry   2
8   Raspberry   6
9   Raspberry   8
10  Raspberry   10

这将绘制以下内容:

df.plot()

enter image description here

我要为这些点(尤其是点之间的线)着色,苹果类别为绿色,蓝莓为蓝色,树莓为红色。

我假设我可以将df分为三个独立的系列,并分别用三种独立的颜色绘制它们,但是肯定有一种更Python化的方式(也许可以像上面建议的那样部署某种色相参数)。

谢谢!

0 个答案:

没有答案