使用pandas数据框绘制多条不同颜色的线。

6 浏览
0 Comments

使用pandas数据框绘制多条不同颜色的线。

我有一个数据框,看起来像下面这样:

   颜色  x   y
0    红  0   0
1    红  1   1
2    红  2   2
3    红  3   3
4    红  4   4
5    红  5   5
6    红  6   6
7    红  7   7
8    红  8   8
9    红  9   9
10  蓝  0   0
11  蓝  1   1
12  蓝  2   4
13  蓝  3   9
14  蓝  4  16
15  蓝  5  25
16  蓝  6  36
17  蓝  7  49
18  蓝  8  64
19  蓝  9  81

我最终想要两条曲线,一条是蓝色的,一条是红色的。红色的曲线应该是y=x,蓝色的曲线应该是y=x^2。

当我执行以下操作时:

df.plot(x='x', y='y')

输出是这样的:

有没有办法让pandas知道有两组数据,并相应地对它们进行分组?我想能够指定列颜色作为数据集的区分因子。

0