Plotly: как сделать сгруппированные гистограммы друг над другом?

Это сюжетный вопрос, так как я хочу иметь интерактивный график. Я пытаюсь сделать два набора сгруппированных гистограмм друг над другом, в которых один из наборов не заполнен, но с толстой линией, а другой набор заполнен.

Это мой фиктивный набор данных:

      from numpy import random
import pandas as pd

list = []
for species in ['C1', 'C2', 'C3', 'C4']:
    for catalyst in ['CHA', 'SAPO', 'MOF','None']:
        for add in [True, False]:
            new_data = {'species': species, 
                        'catalyst': catalyst, 
                        'energy': random.uniform(low=0.0, high=1.0, size=None), 
                        'with_additive': add}
            list.append(new_data)
df = pd.DataFrame(list)            
df

Далее, в сюжете мне удается сделать гистограммы один над другим, как это

      fig = px.bar(df,
             x='species',
             y='energy',
             facet_row ='with_additive',
             color = 'catalyst',
             barmode ='group',
             pattern_shape="with_additive", pattern_shape_sequence=[".", ""]
            )
fig

с этим результатом:

Я также могу сделать их на одном сюжете, но тогда два набора окажутся рядом, как показано ниже:

      fig = px.bar(df,
             x='species',
             y='energy',
             #facet_row ='with_additive',
             color = 'catalyst',
             barmode ='group',
             pattern_shape="with_additive", pattern_shape_sequence=[".", ""]
            )
fig

Производство этого:

Затем я попытался добавить второй набор в качестве трассировки, но не удалось сделать это сгруппированным:

      fig = px.bar(df[df.with_additive==True],
             x='species',
             y='energy',
             #facet_row ='with_additive',
             color = 'catalyst',
             barmode ='group',

            )

fig.add_trace(
    go.Bar(x=df[df.with_additive==True].species, y=df[df.with_additive==True].energy,
           #color = 'surface',
           alignmentgroup ='species',
           base='overlay',
        )
    )

Итак, чего мне не хватает, так это: как получить один из наборов с контуром черной линии и получить, как два набора идеально накладываются друг на друга, как в следующих примерах matplotlib:

      
df_pivot1 = pd.pivot_table(
   df[df.with_additive==True],
   values="energy",
   index="catalyst",
   columns="species",
)

df_pivot2 = pd.pivot_table(
   df[df.with_additive==False],
   values="energy",
   index="catalyst",
   columns="species",
)

 
fig, ax = plt.subplots(figsize=(7,5)) 
df_pivot1.plot(kind="bar", ax=ax)
df_pivot2.plot(kind="bar", edgecolor='black', linewidth=2, ax=ax, fill=False)

handles, labels = ax.get_legend_handles_labels()
handle_list =handles[0:5]
label_list= labels[0:4]
label_list.append('no additive')
plt.legend(handle_list, label_list, loc='upper center', bbox_to_anchor=(0.5, 1.1), ncol=5)

1 ответ

Вы можете сделать это следующим образом:

1. Соберите две фигуры fig1=px.bar()а также fig2=px.bar()

2. Внесите необходимые изменения, чтобы выделить дорожки без цвета и с черной линией:

       fig2.update_traces(marker_color = 'rgba(0,0,0,0)', marker_line_color = 'black')

3. Измените имена трасс на:

      fig2.for_each_trace(lambda t: t.update(name = t.name + ' No additives'))

3. Настройте новую фигуру fig=go.Figure(fig1.data)

4. Добавьте отредактированные трассы в figиз fig2с использованием fig.add_traces(fig2.data)

Вот и все:

Сюжет

Код:

      from numpy import random
import pandas as pd
import plotly.express as px
import plotly.graph_objects as go

list = []
for species in ['C1', 'C2', 'C3', 'C4']:
    for catalyst in ['CHA', 'SAPO', 'MOF','None']:
        for add in [True, False]:
            new_data = {'species': species, 
                        'catalyst': catalyst, 
                        'energy': random.uniform(low=0.0, high=1.0, size=None), 
                        'with_additive': add}
            list.append(new_data)
df = pd.DataFrame(list)            

fig1 = px.bar(df[df.with_additive==True],
             x='species',
             y='energy',
             facet_row ='with_additive',
             color = 'catalyst',
             barmode ='group',
             # pattern_shape="with_additive", pattern_shape_sequence=[".", ""]
            )

fig2 = px.bar(df[df.with_additive==False],
             x='species',
             y='energy',
             facet_row ='with_additive',
             color = 'catalyst',
             barmode ='group',
            )

fig2.update_traces(marker_color = 'rgba(0,0,0,0)', marker_line_color = 'black')
fig2.for_each_trace(lambda t: t.update(name = t.name + ' No addititves'))

fig = go.Figure(fig1.data)
fig.add_traces(fig2.data)
fig.show()
Другие вопросы по тегам