如何添加计数并删除countplot的顶部和右侧脊椎?

2024-09-30 16:30:16 发布

您现在位置:Python中文网/ 问答频道 /正文

如何通过在每个条的顶部添加计数并检索顶部和右侧的线(仅保留x轴和y轴)来改进此图?以一种“灵活”的方式(我的意思是,不仅仅是对于这个图,而且可以轻松地用另一个数据进行复制)

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']);

我试着查看一些帖子,比如How to improve this seaborn countplot?,甚至是文档(https://seaborn.pydata.org/generated/seaborn.countplot.html),但我找不到这些东西的答案


Tags: 数据importiristargetdfdataasnp
1条回答
网友
1楼 · 发布于 2024-09-30 16:30:16

IIUC,试试:

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)

输出:

enter image description here

相关问题 更多 >