Pandas 将数据分组到散景图

数据挖掘 Python 可视化 熊猫
2021-09-17 21:17:37

我无法在 Bokeh 中绘制 Pandas 分组数据。

company_id    company_score    date_submitted    company_region
AA            .07              1/1/2017          NW
AB            .08              1/2/2017          NE
CD            .0003            1/18/2017         NW

我已经成功地将数据分组company_id,并使用 10 天/输入计算了 company_score 的简单移动平均值。

score_by_company = df['company_score'].groupby(df['company_id']).rolling(10).mean()

company_id    index           SMA     
BBG000B9WMF7  7613            NaN
              1911663         NaN
              1911664         NaN
              1911665         NaN
              1911666         NaN
              1911667         NaN
              1911668         NaN
              1911669         NaN
              1911670         NaN
              1911671    0.000002
              1911672    0.000002
              etc.       etc.

如何将这些数据转换为时间序列图,每个图都有company_id自己的线?

我觉得我需要从分组数据中创建一个数据字典,但我不确定这是否是正确的方法。

1个回答

您可以通过循环从groupby...中的组来制作图。

或者这也应该工作..

import matplotlib
import matplotlib.pyplot as plt
df['company_score'].groupby(df['company_id']).rolling(10).mean().unstack().plot()

有关使用 pandas 数据帧进行绘图的更多信息,请参阅此内容,以及groupby-object 上循环的内容。最后这个链接非常有用。