Question

我有一个表，其中的行中有不同的方案，而列中的值与它们相关联

| Scenario   |    A |     B |   C |
|:-----------|-----:|------:|----:|
| Scen1      |  6.5 | 0.125 |  52 |
| Scen2      | 16.5 | 1.125 | 152 |
| Scen3      | 26.5 | 2.125 | 252 |

我要生成堆积的条形图similar to this

即我想要三个小节（每个场景Scen1，Scen2，Scen3一个），每个小节都有三个部分（代表A，B，C的值）

我尝试了以下操作，但是无法弄清楚如何正确地将数据帧传递到px.bar。

import pandas as pd
import plotly.express as px

df = pd.DataFrame()
dict1 = {'A': 6.5, 'B': 0.125, 'C': 52}
series1 = pd.Series(data=dict1, name='Scen1')
df = df.append(series1)
dict2 = {'A': 16.5, 'B': 1.125, 'C': 152}
series2 = pd.Series(data=dict2, name='Scen2')
df = df.append(series2)
dict3 = {'A': 26.5, 'B': 2.125, 'C': 252}
series3 = pd.Series(data=dict3, name='Scen3')
df = df.append(series3)
df.index.name = "Scenario"


fig = px.bar(df, x="Scenario", y=["A","B", "C"], title="Test")
fig.show()

Answer 1

使用DataFrame.melt预处理数据：

df = df.reset_index().melt('Scenario')
print (df)
  Scenario variable    value
0    Scen1        A    6.500
1    Scen2        A   16.500
2    Scen3        A   26.500
3    Scen1        B    0.125
4    Scen2        B    1.125
5    Scen3        B    2.125
6    Scen1        C   52.000
7    Scen2        C  152.000
8    Scen3        C  252.000


import plotly.express as px

fig = px.bar(df, x="Scenario", y="value", color="variable", title="Test")
fig.show()

从数据框创建堆叠的条形图

1 个答案: