python sns.heatmap(cm, annot=True)设置横纵坐标
时间: 2024-02-06 07:12:29 浏览: 143
python seaborn heatmap可视化相关性矩阵实例
5星 · 资源好评率100%
在seaborn库中,heatmap函数可以用来绘制热力图。其中,cm是混淆矩阵,annot=True表示在热力图上显示数值。要设置横纵坐标的标签,可以使用set_xticklabels和set_yticklabels函数,例如:
```
import seaborn as sns
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix
cm = confusion_matrix(y_true, y_pred)
sns.heatmap(cm, annot=True)
plt.xlabel('Predicted labels')
plt.ylabel('True labels')
plt.xticks([0.5, 1.5, 2.5], ['label1', 'label2', 'label3'])
plt.yticks([0.5, 1.5, 2.5], ['label1', 'label2', 'label3'])
plt.show()
```
其中,set_xticklabels和set_yticklabels函数用于设置横纵坐标的标签,第一个参数是标签的位置,第二个参数是标签的名称,这里用列表表示。
阅读全文