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()