尝试在plotly(python)中为每个条形图子图创建统一的颜色比例

2024-05-20 02:31:37 发布

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

我已经创建了一个图表,其中包含8个子地块,对应于一个农场中每个风力涡轮机每年的能源产量。每个子批次对应不同的运行年份。我设法将一个很好的色标应用到每个子图中,但每个色标都有不同的范围(基于每个子图中的数据)

我想制作一个有“全局”色标的图,每个图中的值对应于固定的颜色。我将非常感谢你的建议

def aep_turbine_subplot_fig(years, AEP):

    fig = make_subplots(rows = 4, cols = 2, subplot_titles = years)


    fig.add_trace(go.Bar(x = get_turbine_names(),
                    y = AEP.iloc[0,:],
                    name = '2012',
                    marker = {'color': AEP.iloc[0,:],
                              'colorscale': 'RdBu'}),
                    row = 1, col = 1)

    fig.add_trace(go.Bar(x = get_turbine_names(),
                    y = AEP.iloc[1,:],
                    name = '2013',
                    marker = {'color': AEP.iloc[1,:],
                              'colorscale': 'RdBu'}),
                    row = 1, col = 2)

    fig.add_trace(go.Bar(x = get_turbine_names(),
                    y = AEP.iloc[2,:],
                    name = '2014',
                    marker = {'color': AEP.iloc[2,:],
                              'colorscale': 'RdBu'}),
                    row = 2, col = 1)


    fig.add_trace(go.Bar(x = get_turbine_names(),
                    y = AEP.iloc[3,:],
                    name = '2015',
                    marker = {'color': AEP.iloc[3,:],
                              'colorscale': 'RdBu'}),
                    row = 2, col = 2)


    fig.add_trace(go.Bar(x = get_turbine_names(),
                    y = AEP.iloc[4,:],
                    name = '2016',
                    marker = {'color': AEP.iloc[4,:],
                              'colorscale': 'RdBu'}),
                    row = 3, col = 1)


    fig.add_trace(go.Bar(x = get_turbine_names(),
                    y = AEP.iloc[5,:],
                    name = '2017',
                    marker = {'color': AEP.iloc[5,:],
                              'colorscale': 'RdBu'}),
                    row = 3, col = 2)

    fig.add_trace(go.Bar(x = get_turbine_names(),
                    y = AEP.iloc[6,:],
                    name = '2018',
                    marker = {'color': AEP.iloc[6,:],
                              'colorscale': 'RdBu'}),
                    row = 4, col = 1)

    fig.add_trace(go.Bar(x = get_turbine_names(),
                    y = AEP.iloc[7,:],
                    name = '2019 (Jan to Jun)',
                    marker = {'color': AEP.iloc[7,:],
                              'colorscale': 'RdBu'}),
                    row = 4, col = 2)



    # editing the yaxes in each subplot
    fig.update_yaxes(title_text='AEP [GWh] in 2012', title_font = dict(size = 14), row=1, col=1, range = [0,8.2])
    fig.update_yaxes(title_text='AEP [GWh] in 2013', title_font = dict(size = 14), row=1, col=2, range = [0,8.2])
    fig.update_yaxes(title_text='AEP [GWh] in 2014', title_font = dict(size = 14), row=2, col=1, range = [0,8.2])
    fig.update_yaxes(title_text='AEP [GWh] in 2015', title_font = dict(size = 14), row=2, col=2, range = [0,8.2])
    fig.update_yaxes(title_text='AEP [GWh] in 2016', title_font = dict(size = 14), row=3, col=1, range = [0,8.2])
    fig.update_yaxes(title_text='AEP [GWh] in 2017', title_font = dict(size = 14), row=3, col=2, range = [0,8.2])
    fig.update_yaxes(title_text='AEP [GWh] in 2018', title_font = dict(size = 14), row=4, col=1, range = [0,8.2])
    fig.update_yaxes(title_text='AEP [GWh] in 2019', title_font = dict(size = 14), row=4, col=2, range = [0,8.2])

    # LAYOUT
    fig.update_layout(
            title = 'AEP per turbine',
            xaxis_tickfont_size = 14,
            barmode='group',
            bargap=0.15, # gap between bars of adjacent location coordinates.
            bargroupgap=0.1, # gap between bars of the same location coordinate.
            showlegend = False,
            plot_bgcolor ='rgb(160,160,160)',

        )
    fig.write_image(get_fig_dir() + 'AEP_perTurbine.png', width = 800, height = 800)
    fig.show(renderer = 'png', width = 800, height = 1000)
    return plot(fig, auto_open = True)

The graph I got with the above code. As you may see, each of the subplots has a separate colorscale.


Tags: inaddgosizegettitlefigtrace