Python:如何使用多行绘制图形?

时间:2017-10-20 16:08:17

标签: python pandas pyspark

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

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()

DataFrame代码:

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

欲望结果:

{{1}}

在该示例中,我有3个不同的 id 行,因此图形将用三行绘制,但在实际数据框中, id 的数量将是不知道。

1 个答案:

答案 0 :(得分:1)

如果你有一个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()

enter image description here