Overlaying two histograms with plotly express

I'd like to overlay two histograms which I currently display only one next to the other using the following simplistic code. The two dataframes are not the same length, but it still makes sense to overlay their histogram values.

import plotly.express as px

fig1 = px.histogram(test_lengths, x='len', histnorm='probability', nbins=10)
fig2 = px.histogram(train_lengths, x='len', histnorm='probability', nbins=10)
fig1.show()
fig2.show()

with pure plotly, this is the way, copied from the documentation:

import plotly.graph_objects as go

import numpy as np

x0 = np.random.randn(500)
# Add 1 to shift the mean of the Gaussian distribution
x1 = np.random.randn(500) + 1

fig = go.Figure()
fig.add_trace(go.Histogram(x=x0))
fig.add_trace(go.Histogram(x=x1))

# Overlay both histograms
fig.update_layout(barmode='overlay')
# Reduce opacity to see both histograms
fig.update_traces(opacity=0.75)
fig.show()

I just wonder if there's any particularly idiomatic way with plotly express. Hopefully this also works to exeplify the completeness and different levels of abstraction between plotly and plotly express.

Solution 1:

The trick is to make a single Plotly Express figure by combining the data into a tidy dataframe, rather than to make two figures and try to combine them (which is currently impossible):

import numpy as np
import pandas as pd
import plotly.express as px

x0 = np.random.randn(250)
# Add 1 to shift the mean of the Gaussian distribution
x1 = np.random.randn(500) + 1

df =pd.DataFrame(dict(
    series=np.concatenate((["a"]*len(x0), ["b"]*len(x1))), 
    data  =np.concatenate((x0,x1))
))

px.histogram(df, x="data", color="series", barmode="overlay")

Yields:

Solution 2:

You can get at the px structure and use it to create a figure. I had a desire to show a stacked histogram using the 'color' option that's in express but hard to re-create in pure plotly.

Given a dataframe (df) with utctimestamp as a time index, severity and category as things to count in the histogram I used this to get a stacked histogram:

figure_data=[]
figure_data.extend([i for i in px.histogram(df, x="utctimestamp", color="severity", histfunc="count").to_dict()['data']])
figure_data.extend([i for i in px.histogram(df, x="utctimestamp", color="category", histfunc="count").to_dict()['data']])
fig=go.Figure(figure_data)
fig.update_layout(barmode='stack')
fig.update_traces(overwrite=True, marker={"opacity": 0.7}) 
fig.show()

tl;dr px.histogram creates a list of histogram objects that you can grab as a list and render via go.Figure.

I can't post inline, but here's stacked histograms from px https://imgur.com/a/l7BblZo

Solution 3:

If one desires to use plotly's graph_objects module one can instead use barmode="overlay" as shown below for 2 histograms.

import plotly.graph_objects as go
fig = go.Figure(data=[go.Histogram(x=x)])
fig.add_trace(go.Histogram(x=x,))
fig.update_layout(barmode='overlay')