pythonpandasseabornjointplot

How to overlay a Seaborn jointplot with a "marginal" (distribution histogram) from a different dataset


I have plotted a Seaborn JointPlot from a set of "observed counts vs concentration" which are stored in a pandas DataFrame. I would like to overlay (on the same set of axes) a marginal (ie: univariate distribution) of the "expected counts" for each concentration on top of the existing marginal, so that the difference can be easily compared.

This graph is very similar to what I want, although it will have different axes and only two datasets:

Here is an example of how my data is laid out and related:

df_observed

x axis--> log2(concentration): 1,1,1,2,3,3,3 (zero-counts have been omitted)

y axis--> log2(count): 4.5, 5.7, 5.0, 9.3, 16.0, 16.5, 15.4 (zero-counts have been omitted)

df_expected

x axis--> log2(concentration): 1,1,1,2,2,2,3,3,3

an overlaying of the distribution of df_expected on top of that of df_observed would therefore indicate where there were counts missing at each concentration.

What I currently have

Jointplot with the observed counts at each concentration Separate jointplot of the expected counts at each concentration. I want the marginal from this plot to be overlaid on top of the marginal from the above jointplot

PS: I am new to Stack Overflow so any suggestions about how to better ask questions will be met with gratitude. Also, I have searched extensively for an answer to my question but to no avail. In addition, a Plotly solution would be equally helpful. Thank you


Solution

  • Whenever I try to modify a JointPlot more than for what it was intended for, I turn to a JointGrid instead. It allows you to change the parameters of the plots in the marginals.

    Below is an example of a working JointGrid where I add another histogram for each marginal. These histograms represent the expected value that you wanted to add. Keep in mind that I generated random data so it probably doesn't look like yours.

    enter image description here

    Take a look at the code, where I altered the range of each second histogram to match the range from the observed data.

    import pandas as pd
    import numpy as np
    import seaborn as sns
    import matplotlib.pyplot as plt
    
    df = pd.DataFrame(np.random.randn(100,4), columns = ['x', 'y', 'z', 'w'])
    
    plt.ion()
    plt.show()
    plt.pause(0.001)
    
    p = sns.JointGrid(
        x = df['x'],
        y = df['y']
        )
    
    p = p.plot_joint(
        plt.scatter
        )
    
    p.ax_marg_x.hist(
        df['x'],
        alpha = 0.5
        )
    
    p.ax_marg_y.hist(
        df['y'],
        orientation = 'horizontal',
        alpha = 0.5
        )
    
    p.ax_marg_x.hist(
        df['z'],
        alpha = 0.5,
        range = (np.min(df['x']), np.max(df['x']))
        )
    
    p.ax_marg_y.hist(
        df['w'],
        orientation = 'horizontal',
        alpha = 0.5,
        range = (np.min(df['y']), np.max(df['y'])),
        )
    

    The part where I call plt.ion plt.show plt.pause is what I use to display the figure. Otherwise, no figure appears on my computer. You might not need this part.

    Welcome to Stack Overflow!