matplotlib 标记目标变量

camsedfj 于 2022-11-15 发布在其他

关注(0)|答案(1)|浏览(113)

我试图绘制心脏病与性别的关系图。在图中，我试图将0标记为不存在，将1标记为存在。有什么方法可以做到这一点吗？我尝试了以下代码并绘制了图：

ax = sns.countplot(x='target', hue='sex', data=heartt)
plt.show()

1条答案

如果我是你，我不会重新标记坐标轴的刻度，除非你非常确定seborn将如何处理所有变量的顺序。在数据中这样做更安全：

heartt['target_label'] = heartt['target'].map({0: 'absent', 1: 'present'})
ax = sns.countplot(x='target_label', hue='sex', data=heartt)
plt.show()