Matplotlib数据可视化

入门·10 分钟·通过可运行代码讲解「Matplotlib数据可视化」,覆盖环境准备、核心代码、参数说明、异常处理、工程化改造和练习任务,适合 AI 编程入门。

Matplotlib数据可视化

Matplotlib 是 Python 中经典的数据可视化库。它可以把数据变成折线图、柱状图、散点图等,帮助你理解趋势、分布和异常。

为什么AI学习需要可视化

只看表格很难发现问题。比如训练损失是否下降、某类样本是否特别少、预测结果是否偏向某个区间,这些都适合画图观察。

画折线图

import matplotlib.pyplot as plt
 
epochs = [1, 2, 3, 4, 5]
loss = [1.2, 0.9, 0.7, 0.62, 0.58]
 
plt.plot(epochs, loss, marker='o')
plt.title('Training Loss')
plt.xlabel('Epoch')
plt.ylabel('Loss')
plt.grid(True, alpha=0.3)
plt.show()

训练过程中,loss 曲线能帮助你判断模型是否还在学习。

画柱状图

categories = ['positive', 'neutral', 'negative']
counts = [120, 80, 40]
 
plt.bar(categories, counts)
plt.title('Label Distribution')
plt.show()

分类任务中,先看标签分布很重要。如果某一类样本太少,模型可能学不好。

画散点图

hours = [1, 2, 3, 4, 5]
scores = [55, 60, 68, 75, 82]
 
plt.scatter(hours, scores)
plt.xlabel('Study Hours')
plt.ylabel('Score')
plt.title('Study Time vs Score')
plt.show()

散点图适合观察两个变量之间是否有关联。

保存图片

plt.plot(epochs, loss)
plt.savefig('loss.png', dpi=150, bbox_inches='tight')

写实验报告或 README 时,保存图片比只截图更规范。

中文显示问题

plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False

不同系统字体不同,如果乱码,需要换成本机已有中文字体。

小结

Matplotlib 的价值在于帮助你看见数据。AI 项目中,很多问题不是模型太弱,而是数据分布、标签质量或训练过程出了问题。