如何在图中每年显示一根 plot

Question

再会，

我有一个简单的问题，我似乎无法得到解决方案。 我有一个数据框，其中包含每年参加特定 class 的学生人数。 我想制作一个条形图，其中 x 轴上的 class id 和 y 轴上的学生人数，范围 slider 一次只显示一年。

这是我的数据：

在：

import pandas as pd
import plotly.graph_objs as go


d = {'year': [2017, 2017, 2017, 2018, 2018, 2018, 2019, 2019, 2019, 2020, 2020, 2020, 2021, 2021, 2021, 2022, 2022, 2022,], 
     'class_id': [101, 102, 103, 101, 102, 103, 102, 103, 105, 105, 106, 107, 106, 107, 108, 106, 107, 108], 
     'num_students': [24, 23, 11, 35, 22, 16, 36, 33, 33, 23, 37, 36, 37, 43, 37, 39, 29, 11]}
df = pd.DataFrame(data=d)

出去：

    year    class_id    num_students
0   2017    101         24
1   2017    102         23
2   2017    103         11
3   2018    101         35
4   2018    102         22
5   2018    103         16
6   2019    102         36
7   2019    103         33
8   2019    105         33
9   2020    105         23
10  2020    106         37
11  2020    107         36
12  2021    106         37
13  2021    107         43
14  2021    108         37
15  2022    106         39
16  2022    107         29
17  2022    108         11

这是图表：

data = [go.Bar(
   x = df.class_id,
   y = df.num_students
)]
fig = go.Figure(data=data)
fig.show()

现在我需要添加一个 slider，它一次只显示一年的数据，而不是包含所有数据的堆积条形图，但我似乎无法正确处理。

非常感谢任何和所有的帮助，谢谢！

Answer 1

首先，这是您的 dataframe：

import pandas as pd
import plotly.graph_objs as go

d = {'year': [2017, 2017, 2017, 2018, 2018, 2018, 2019, 2019, 2019, 2020, 2020, 2020, 2021, 2021, 2021, 2022, 2022, 2022,], 
     'class_id': [101, 102, 103, 101, 102, 103, 102, 103, 105, 105, 106, 107, 106, 107, 108, 106, 107, 108], 
     'num_students': [24, 23, 11, 35, 22, 16, 36, 33, 33, 23, 37, 36, 37, 43, 37, 39, 29, 11]}
df = pd.DataFrame(data=d)
df['class_id'] = df['class_id'].astype(str)

这是绘图部分：

fig = go.Figure()

for y in df.year.unique():
    fig.add_trace(go.Bar(x=df.query('year==@y')['class_id'],
                        y=df.query('year==@y')['num_students'], name=str(y)))

fig.update_layout(
    updatemenus=[
        {
            "type": "buttons",
            "showactive":True,
            "active":0,
            "buttons": [
                {
                    "label": str(y1),
                    "method": "update",
                    "args": [
                        {"visible": [y1 == y2 for y2 in df.year.unique()]}
                    ],
                }
                for y1 in df.year.unique()
            ]
        }
    ]
)

fig.for_each_trace(
    lambda trace: trace.update(visible=False) if trace.name != str(df.year.unique()[0]) else ()
)

  
fig.show()