在Plotly直方图中,如何使每个动画帧成为数据帧中的一行?

2024-04-25 19:37:04 发布

您现在位置:Python中文网/ 问答频道 /正文

我有以下数据帧:

       month  stories  comments  comment_authors  story_authors
0    2006-10       49        12                4             16
1    2007-02      564       985              192            163
2    2007-03     1784      4521              445            287

我试图构建一个绘图直方图,其中有四个分类箱(x轴),分别对应于storiescommentscomment_authorsstory_authors列,计数(y轴)是特定month(即特定行)的给定数量。然后,我尝试使用animation_frameanimation_group根据Plotly Express中的月份设置直方图的动画

例如,month=2006-10的第一个直方图如下所示:

50 |  ____
45 | |    |
40 | |    |
35 | |    |
30 | |    |
25 | |    |
20 | |    |
15 | |    |                               ____
10 | |    |    ____                      |    |
 5 | |    |   |    |    ______           |    |
 0  ----------------------------------------------------
     stories  comments  comment_authors  story_authors

在下一个动画帧的直方图中,它将从storiescommentscomment_authorsstory_authors列中读取month=2007-02的值

这是否可以在Plotly中构建?有没有比px.Histogram更好的数字可以使用,比如px.Bar?我尝试将列放在x轴上,并使用月份作为动画帧,但这会将列堆叠到直方图上的一个箱子中,并使用整个列的计数,而不是特定行的值

histogram = dcc.Graph(figure=px.histogram(
    df, x=['stories', 'comments', 'comment_authors', 'story_authors'],
    animation_frame='month', animation_group='month'
))

Tags: groupcomment动画plotly直方图framecommentsauthors
2条回答

@r-初学者的答案是最好的方法,但我想与大家分享如何手动操作,以防将来有人发现这个问题(这样我的辛劳就不会白费了)。下面是在@r-初学者回答之前我是如何手动完成的

# build list of frames for animation
cols = ['stories', 'comments', 'comment_authors', 'story_authors']
frames = list(map(
    lambda index: go.Frame(
        data=[go.Bar(
            x=cols,
            y=list(map(lambda col: df.iloc[index][col], cols))
        )]
    ), range(len(df.index))
))

# compute chart height
m = max(list(filter(
    lambda m: not isinstance(m, str), list(df.max())
))) * 1.1

# compute animation steps for slider
steps = list(map(
    lambda index: {
        'args': [
            [str(df.iloc[index]['month'])],
            {
                'frame': {'duration': 300, 'redraw': False},
                'mode': 'immediate',
                'transition': {'duration': 300}
            }
        ],
        'label': str(df.iloc[index]['month']),
        'method': 'animate'
    }, range(len(df.index))
))

# metadata for slider element
sliders_dict = {
    "active": 0,
    "yanchor": "top",
    "xanchor": "left",
    "currentvalue": {
        "font": {"size": 20},
        "prefix": "Month: ",
        "visible": True,
        "xanchor": "right"
    },
    "transition": {"duration": 300, "easing": "cubic-in-out"},
    "pad": {"b": 10, "t": 50},
    "len": 0.9,
    "x": 0.1,
    "y": 0,
    "steps": steps
}

# histogram figure - actually a bar chart
histogram = dcc.Graph(id='counts-histogram', figure=go.Figure(
    data=frames[0].data[0],
    layout=go.Layout(
        yaxis=dict(range=[0, m], autorange=False),
        updatemenus=[
            {
                "buttons": [
                    {
                        "args": [None, {"frame": {"duration": 500, "redraw": False},
                                        "fromcurrent": True, "transition": {"duration": 300,
                                                                            "easing": "quadratic-in-out"}}],
                        "label": "Play",
                        "method": "animate"
                    },
                    {
                        "args": [[None], {"frame": {"duration": 0, "redraw": False},
                                          "mode": "immediate",
                                          "transition": {"duration": 0}}],
                        "label": "Pause",
                        "method": "animate"
                    }
                ],
                "direction": "left",
                "pad": {"r": 10, "t": 87},
                "showactive": False,
                "type": "buttons",
                "x": 0.1,
                "xanchor": "right",
                "y": 0,
                "yanchor": "top"
            }
        ],
        sliders = [sliders_dict]
    ),
    frames=frames
))

不可能用您呈现的数据绘制直方图。你能做的最好的是一个条形图,你可以用一个时间序列来制作它的动画。我已经修改了官方reference中的样本以适合您的数据

import pandas as pd
import numpy as np
import io

data = '''
      month  stories  comments  comment_authors  story_authors
0    2006-10       49        12                4             16
1    2007-02      564       985              192            163
2    2007-03     1784      4521              445            287
'''

df = pd.read_csv(io.StringIO(data), delim_whitespace=True)
df.set_index('month', inplace=True)
dfs = df.unstack().reset_index()
dfs.columns = ['category', 'month', 'value']
import plotly.express as px

fig = px.bar(dfs, x='category', y='value', color='category', animation_frame='month', range_y=[0,dfs['value'].max()])

fig.show()

enter image description here

相关问题 更多 >