如何在 streamlit 上按周期分组绘制图表？

Posted 2023-03-29

【中文标题】如何在 streamlit 上按周期分组绘制图表？

【英文标题】：How can I make a plotly graph on streamlit grouped by period?

【发布时间】：2021-05-12 02:04:27

【问题描述】：

我正在尝试使用 Plotly + Streamlit 绘制折线图，​​但按月对数据进行分组。

我有这些数据：

id      date          grade
A       2020-01-01    100
B       2020-01-11    200
C       2020-01-21    500
D       2020-01-21    300
E       2020-02-01    100
F       2020-02-01    200

我想按月绘制平均成绩。这是我到目前为止得到的：

df = pd.to_datetime(df["date"], format="%m/%d/%y %I:%M%p")

df_grouped = df.groupby(
    by=[df.index.month, df.index.year]
)

df.grade.mean().plot()

问题是我不知道如何使用 plotly 和 streamlit 来做同样的事情（或类似的事情）。

有什么想法吗？

【参考方案1】：

你还没有指定你想要什么样的酒吧类型，但我猜go.Bar() 会做。但我们可以改变这一点。对于分组和聚合，我将使用以下方法：

df['date'] = pd.to_datetime(df["date"])
df['months'] = df['date'].dt.month_name()
df_months = df.groupby(['months']).agg('mean').reset_index()
new_order = ['January', 'February', 'March', 'April', 'May', 'June', 'July', 'August', 'September', 'October', 'November', 'December']
df_months['months'] = pd.Categorical(df_months['months'], categories=new_order, ordered=True)
df_months = df_months.sort_values('months')

为什么这么复杂？因为我假设您希望在 x 轴上有月份名称。在按月份名称对值进行分组和聚合之后，月份的顺序可能会变得混乱。上面有些费力的方法确保不会发生这种情况，并且您可以按照正确的月份顺序绘制此条形图：

完整代码：

import pandas as pd
import plotly.graph_objects as go

df = pd.DataFrame('id': 0: 'A', 1: 'B', 2: 'C', 3: 'D', 4: 'E', 5: 'F',
                     'date': 0: '2020-01-01',
                      1: '2020-01-11',
                      2: '2020-01-21',
                      3: '2020-01-21',
                      4: '2020-02-01',
                      5: '2020-02-01',
                     'grade': 0: 100, 1: 200, 2: 500, 3: 300, 4: 100, 5: 200)

df['date'] = pd.to_datetime(df["date"])
df['months'] = df['date'].dt.month_name()
df_months = df.groupby(['months']).agg('mean').reset_index()
new_order = ['January', 'February', 'March', 'April', 'May', 'June', 'July', 'August', 'September', 'October', 'November', 'December']
df_months['months'] = pd.Categorical(df_months['months'], categories=new_order, ordered=True)
df_months = df_months.sort_values('months')

fig=go.Figure(go.Bar(x=df_months.months, y=df_months.grade))
fig.show()

【讨论】：

我正在尝试这段代码，但有一件事让我想到，如果我有多年的时间，这应该行不通，对吧？

@dummyds 也许不是这种确切的方法。但可能还有其他方法。所以我建议你把它写下来作为答案，我们会从那里拿走它