Python:如何用多条线绘制图形?

泰国

我有一个包含三列的数据框。我需要用多条线绘制图形。在该图中,每条线代表一个id,x 轴代表月份列,y 轴代表qtd列。

+-----+-----+----+
|month| id  |qtd |
+-----+-----+----+
| 1   |  1  | 1  |
| 1   |  2  | 3  |
| 1   |  3  | 6  |
| 2   |  4  | 2  |
| 2   |  5  | 3  |
| 3   |  6  | 4  |
| 3   |  7  | 5  |
| 3   |  8  | 4  |
+-----+-----+----+

数据帧代码:

l=[(1,1,1),(1,2,3),(1,3,6),(2,4,2),(2,5,3),(3,6,4),(3,7,5),(3,8,4)]
names=["month","id","qtd"]
db=sqlContext.createDataFrame(l,names)
db.show()

想要的结果:

plot with x=qtd, y=month and one line for each id line

在那个例子中,我有 3 条不同的id线,所以图形将用三条线绘制,但在实际数据框中,id数量是未知的。

咕噜噜

如果您有 Pandas 数据框,您可以按id如下方式分组并绘制每条线:

fig, ax = plt.subplots()

for key, grp in df.groupby('id'):
    ax = grp.plot(ax=ax, x='qtd', y='month', label=key)

fig.show()

在此处输入图片说明

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章