CODEWITHSUNDEEP
pythonpandasmatplotlibseabornviolin-plot
Chang Ye
Chang Ye

Reputation: 1215

How to assign different position for each group in a violinplot

The shape of violin plot is useful for visualizing data distribution of grouped data. The size of each group can also be visualized as the area of the 'violin'.

But when the data is heterogeneous， the width of certain group is too small to show any meaning info (Fri group in Figure 1). There is width option in seaborn.violinplot for enlarging the size of the plot.

However, once group of small size is enlarged into a suitable scale, the large ones will become 'too large'(Sat group in Figure 2) and overlap with each other.

Thus, my question is how to assign different gaping distance for violin plot in seaborn.

The demo

Code for generating the Figure 1:

import seaborn as sns

tips = sns.load_dataset("tips")
ax = sns.violinplot(x="day", y="total_bill", hue="sex",
                    data=tips, palette="Set2", split=True,
                    scale="count", inner="stick",
                    scale_hue=False, bw=.2)

Figure 1

fig1

Code for generating the Figure 2:

import seaborn as sns

tips = sns.load_dataset("tips")
ax = sns.violinplot(x="day", y="total_bill", hue="sex",
                    data=tips, palette="Set2", split=True,
                    scale="count", inner="stick", width=2.5
                    scale_hue=False, bw=.2)

Figure 2

fig2

What is your solution?

tips["day_n"] = tips["day"].map(dict(zip(tips["day"].unique(), [1, 2, 4, 6])))

But it seems that seaborn does not support numeric data, the distance between group keep just unchanged or mess up, when switching the x, y axis.

Code for generating the Figure 3:

ax = sns.violinplot(y="day_n", x="total_bill", hue="sex",
                    data=tips, palette="Set2", split=True,
                    scale="count", inner="stick", width=2.5,
                    scale_hue=False, bw=.2)

Figure 3

fig3

Upvotes: 6

Views: 3917

Answers (1)

Ulrich Stern
Ulrich Stern

Reputation: 11301

Using the order parameter can achieve the [1, 2, 4, 6] positions on the x-axis:

import seaborn as sns, matplotlib.pyplot as plt

tips = sns.load_dataset("tips")
ax = sns.violinplot(x="day", y="total_bill", hue="sex",
                    data=tips, palette="Set2", split=True,
                    scale="count", inner="stick",
                    scale_hue=False, bw=.2, width=2.5,
                    order=('Thur', 'Fri', '', 'Sat', '', 'Sun'))

# get rid of ticks for empty columns (levels)
ax.set_xticks([0,1,3,5])
ax.set_xticklabels(['Thur', 'Fri', 'Sat', 'Sun'])

plt.show()

Here the result:violinplot

Upvotes: 6

Related Questions