Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to create spacing between same subgroup in seaborn boxplot?

I currently have a seaborn box plot that looks like this: Current box plot

Each grouping ('hue') on the x axis are all touching each other.

The code for this boxplot is this:

bp_all = sns.boxplot(x='X_Values', y='Y_values', hue='Groups123', data=mydataframe, width=0.8, showfliers=False, linewidth=4.5, palette='coolwarm')

Is there any way to create a small space between the 3 groups so that they are not touching each other?

like image 471
Eric Avatar asked Jul 01 '19 15:07

Eric


1 Answers

I found a solution posted by another user. This function is used to adjust the width of all the objects in your created figure by a factor of your choosing

from matplotlib.patches import PathPatch

def adjust_box_widths(g, fac):
    """
    Adjust the withs of a seaborn-generated boxplot.
    """

    # iterating through Axes instances
    for ax in g.axes:

        # iterating through axes artists:
        for c in ax.get_children():

            # searching for PathPatches
            if isinstance(c, PathPatch):
                # getting current width of box:
                p = c.get_path()
                verts = p.vertices
                verts_sub = verts[:-1]
                xmin = np.min(verts_sub[:, 0])
                xmax = np.max(verts_sub[:, 0])
                xmid = 0.5*(xmin+xmax)
                xhalf = 0.5*(xmax - xmin)

                # setting new width of box
                xmin_new = xmid-fac*xhalf
                xmax_new = xmid+fac*xhalf
                verts_sub[verts_sub[:, 0] == xmin, 0] = xmin_new
                verts_sub[verts_sub[:, 0] == xmax, 0] = xmax_new

                # setting new width of median line
                for l in ax.lines:
                    if np.all(l.get_xdata() == [xmin, xmax]):
                        l.set_xdata([xmin_new, xmax_new])

For example:

fig = plt.figure(figsize=(15, 13))
bp = sns.boxplot(#insert data and everything)
adjust_box_widths(fig, 0.9)

Example figure

like image 172
Eric Avatar answered Sep 23 '22 13:09

Eric