当前位置:首页>python>python数据可视化技巧的100个练习 --34. 混淆矩阵的矩阵图

python数据可视化技巧的100个练习 --34. 混淆矩阵的矩阵图

  • 2026-09-06 19:35:42
python数据可视化技巧的100个练习 --34. 混淆矩阵的矩阵图
重要性★★★★☆
难度★★★☆☆
您是一家零售公司的数据分析师。您的任务是评估一个分类模型的性能,该模型根据客户的浏览历史预测客户是否会购买商品。
为此,您需要创建一个混淆矩阵,并使用矩阵图将其可视化。
生成实际值和预测值的样本数据集,然后创建混淆矩阵的矩阵图。
确保您的代码正确输出图表。

【数据生成代码示例】

import numpy as npnp.random.seed(42)actual = np.random.choice([0, 1], size=100)predicted = np.random.choice([0, 1], size=100)

【图表答案】

【代码答案】

import numpy as npimport matplotlib.pyplot as pltimport seaborn as snsfrom sklearn.metrics import confusion_matrixnp.random.seed(42)actual = np.random.choice([0, 1], size=100)predicted = np.random.choice([0, 1], size=100)cm = confusion_matrix(actual, predicted)plt.figure(figsize=(8, 6))sns.heatmap(cm, annot=True, fmt='d', cmap='Blues', xticklabels=['Predicted No', 'Predicted Yes'], yticklabels=['Actual No', 'Actual Yes'])plt.xlabel('Predicted')plt.ylabel('Actual')plt.title('Confusion Matrix')plt.show()
要解决此问题,您首先需要生成实际值和预测值的样本数据集。
可以使用 NumPy 的 random choice 函数创建二进制值(0 和 1)的数组。
一旦有了数据,就可以使用 sklearn.metrics 模块中的 confusion_matrix 函数计算混淆矩阵。
混淆矩阵是一个通常用于描述分类模型性能的表格。
它显示了真正例、真负例、假正例和假负例的计数。
接下来,要可视化混淆矩阵,您可以使用 seaborn 库,它提供了用于绘制吸引人的统计图形的高级接口。
seaborn 的 heatmap 函数可用于创建混淆矩阵的矩阵图。
在图表中,您可以使用 annot 参数显示每个单元格中的计数,使用 cmap 参数选择颜色映射。
可以使用 xticklabels 和 yticklabels 参数设置 x 轴和 y 轴的标签。
最后,您可以分别使用 plt.xlabel、plt.ylabel 和 plt.title 为图表添加标签和标题。
使用 plt.show 函数显示图表。

【小知识】

  • 混淆矩阵是评估分类模型性能的关键工具,特别是在处理不平衡数据集时。
  • 真正例(TP)和真负例(TN)分别是模型正确预测正类和负类的情况。
  • 假正例(FP)和假负例(FN)分别是模型错误预测正类和负类的情况。
  • 可以使用混淆矩阵计算其他重要指标,如准确率、精确率、召回率和 F1 分数。
  • Seaborn 的 heatmap 函数提供了一种简单而强大的方式来可视化混淆矩阵,使结果更容易解读。

最新文章

随机文章