How to add counts and remove top and right spines of a countplot?

Question

How can I improve this graph by adding the count on the top of each bar and retrieving the lines on the top and on the right (just keeping the x and y axis)? In a "flexible" way (I mean, not just for this graph, but easily to replicate with another data)

import numpy as np
import pandas as pd
from sklearn.datasets import load_iris

iris = load_iris()
df = pd.DataFrame(data= np.c_[iris['data'], iris['target']],
                 columns= iris['feature_names'] + ['target'])
sns.countplot(df['target']);

I tried looking at some posts, like How to improve this seaborn countplot?, and even at the documentation (https://seaborn.pydata.org/generated/seaborn.countplot.html), but I couldn't find a these things.

Similar to [this post](https://stackoverflow.com/a/31754317/12046409), but leaving out the division by `total` — JohanC, Jul 22 '20 at 17:23
To remove the spines: [How can I remove the top and right axis in matplotlib?](https://stackoverflow.com/questions/925024/how-can-i-remove-the-top-and-right-axis-in-matplotlib) — JohanC, Jul 22 '20 at 17:26

Scott Boston · Accepted Answer · 2020-07-22T18:06:56.987

1

IIUC, Try:

import numpy as np
import pandas as pd
from sklearn.datasets import load_iris

iris = load_iris()
df = pd.DataFrame(data= np.c_[iris['data'], iris['target']],
                 columns= iris['feature_names'] + ['target'])
ax = sns.countplot(df['target']);
for p in ax.patches:
    height = p.get_height()
    ax.text(p.get_x()+p.get_width()/2.,
            height + 3,
            f'{p.get_height()}',
            ha="center") 
    
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)

Output:

edited Jul 22 '20 at 18:06

answered Jul 22 '20 at 17:41

Scott Boston

133,446
13
126
161

Where does this 'height' comes from? I tried running the script and it didin't recognize this variable – Dumb ML Jul 22 '20 at 18:06
1

@DumbML Oops. I removed a line by accident. Updated answer. – Scott Boston Jul 22 '20 at 18:07
Great, many thanks! Just one more thing, if it's not too much, how can I increase the size of the bar labels? How can I make 50 greater? plt.xticks? – Dumb ML Jul 22 '20 at 18:46
1

Try: `ax.tick_params(axis='x', labelsize=50)` – Scott Boston Jul 22 '20 at 18:51
Actually, that only worked for the x-label. But the "50" above the bar are still very small here – Dumb ML Jul 22 '20 at 19:27
1

`ax.text(p.get_x()+p.get_width()/2., height + 3, f'{p.get_height()}', ha="center", fontsize=50)` add fontsize parameter to ax.text method. – Scott Boston Jul 22 '20 at 19:37

How to add counts and remove top and right spines of a countplot?

1 Answers1