plotly-python - 在 Plotly 图形对象中制作颜色图例

Question

我有以下两个代码及其输出。一个是在图形对象中完成的，另一个是使用 plotly express 完成的。如您所见，“go”中的没有图例，“px”中的没有单独的列宽。那么我怎样才能得到第一个的图例，或者修复另一个的宽度呢？

import plotly.graph_objects as go
import pandas as pd

df = pd.DataFrame({'PHA': [451, 149, 174, 128, 181, 175, 184, 545, 131, 106, 1780, 131, 344, 624, 236, 224, 178, 277, 141, 171, 164, 410],
                   'PHA_cum': [451, 600, 774, 902, 1083, 1258, 1442, 1987, 2118, 2224, 4004, 4135, 4479, 5103, 5339, 5563, 5741, 6018, 6159, 6330, 6494, 6904],
                   'trans_cost_cum': [0.14, 0.36, 0.6, 0.99, 1.4, 2.07, 2.76, 3.56, 4.01, 4.5, 5.05, 5.82, 5.97, 6.13, 6.33, 6.53, 6.65, 6.77, 6.9, 7.03, 7.45, 7.9],
                   'Province': ['East', 'East', 'East', 'East', 'East', 'Lapland', 'Lapland', 'Lapland', 'Oulu', 'Oulu', 'Oulu', 'Oulu', 'South', 'South', 'South', 'South', 'West', 'West', 'West', 'West', 'West', 'West'],
                   })

col_list = {'South': 'rgb(222,203,228)',
            'West': 'rgb(204,235,197)',
            'East': 'rgb(255,255,204)',
            'Oulu': 'rgb(179,205,227)',
            'Lapland': 'rgb(254,217,166)'}

provs = df['Province'].to_list()

colors = [col_list.get(item, item) for item in provs]

fig = go.Figure(data=[go.Bar(
    x=df['PHA_cum']-df['PHA']/2,
    y=df['trans_cost_cum'],
    width=df['PHA'],
    marker_color=colors
)])

fig.show()

import plotly.express as px
import pandas as pd

df = pd.DataFrame({'PHA': [451, 149, 174, 128, 181, 175, 184, 545, 131, 106, 1780, 131, 344, 624, 236, 224, 178, 277, 141, 171, 164, 410],
                   'PHA_cum': [451, 600, 774, 902, 1083, 1258, 1442, 1987, 2118, 2224, 4004, 4135, 4479, 5103, 5339, 5563, 5741, 6018, 6159, 6330, 6494, 6904],
                   'trans_cost_cum': [0.14, 0.36, 0.6, 0.99, 1.4, 2.07, 2.76, 3.56, 4.01, 4.5, 5.05, 5.82, 5.97, 6.13, 6.33, 6.53, 6.65, 6.77, 6.9, 7.03, 7.45, 7.9],
                   'Province': ['East', 'East', 'East', 'East', 'East', 'Lapland', 'Lapland', 'Lapland', 'Oulu', 'Oulu', 'Oulu', 'Oulu', 'South', 'South', 'South', 'South', 'West', 'West', 'West', 'West', 'West', 'West'],
                   })

fig = px.bar(df,
             x=df['PHA_cum']-df['PHA']/2,
             y=df['trans_cost_cum'],
             color="Province",
             color_discrete_sequence=px.colors.qualitative.Pastel1
             )
fig.show()

标签： plotly-python

Answer 1

使用graph_objects您需要将每个Province作为跟踪传递，以便填充图例。见下文，唯一真正的变化是循环遍历每个Province.

df = pd.DataFrame({'PHA': [451, 149, 174, 128, 181, 175, 184, 545, 131, 106, 1780, 131, 344, 624, 236, 224, 178, 277, 141, 171, 164, 410],
                   'PHA_cum': [451, 600, 774, 902, 1083, 1258, 1442, 1987, 2118, 2224, 4004, 4135, 4479, 5103, 5339, 5563, 5741, 6018, 6159, 6330, 6494, 6904],
                   'trans_cost_cum': [0.14, 0.36, 0.6, 0.99, 1.4, 2.07, 2.76, 3.56, 4.01, 4.5, 5.05, 5.82, 5.97, 6.13, 6.33, 6.53, 6.65, 6.77, 6.9, 7.03, 7.45, 7.9],
                   'Province': ['East', 'East', 'East', 'East', 'East', 'Lapland', 'Lapland', 'Lapland', 'Oulu', 'Oulu', 'Oulu', 'Oulu', 'South', 'South', 'South', 'South', 'West', 'West', 'West', 'West', 'West', 'West'],
                   })

col_list = {'South': 'rgb(222,203,228)',
            'West': 'rgb(204,235,197)',
            'East': 'rgb(255,255,204)',
            'Oulu': 'rgb(179,205,227)',
            'Lapland': 'rgb(254,217,166)'}

#provs = df['Province'].to_list()

#colors = [col_list.get(item, item) for item in provs]

fig = go.Figure() 

for p in df['Province'].unique(): 
    dat = df[df.Province == p]
    fig.add_trace(go.Bar(
        name = p,
        x=dat['PHA_cum']-dat['PHA']/2,
        y=dat['trans_cost_cum'],
        width=dat['PHA'],
        marker_color= col_list[p]
    ))

fig.show()