matplotlib
Matplotlib
一、Matplotlib快速入门
学习目标
- 了解什么是matplotlib
- 为什么要学习matplotlib
- matplotlib简单图形的绘制
什么是Matplotlib
官网地址：https://matplotlib.org/stable/gallery/index#
是专门用于开发2D图表(包括3D图表)
以渐进、交互式方式实现数据可视化
为什么要学习Matplotlib
可视化是在整个数据挖掘的关键辅助工具，可以清晰的理解数据，从而调整我们的分析方法。
- 能将数据进行可视化,更直观的呈现
- 使数据更加客观、更具说服力
例如下面两个图为数字展示和图形展示：
实现一个简单的Matplotlib画图 — 以折线图为例【掌握】
matplotlib.pyplot模块
matplotlib.pytplot包含了一系列类似于matlab的画图函数。
import matplotlib.pyplot as plt
图形绘制流程：
- 1.【可选】创建画布 – plt.figure()
plt.figure(figsize=(), dpi=)
figsize:指定图的长宽
dpi:图像的清晰度，该值越大图片越清晰，一般设置为100-300之间
返回fig对象
- 2.绘制图像 – plt.plot(x, y)
以折线图为例
- 3.显示图像 – plt.show()
折线图绘制与显示
举例：展现上海一周的天气,比如从星期一到星期日的天气温度如下
"""
pip list | findstr matplotlib
pip uninstall -y matplotlib
pip install matplotlib==3.8.4
"""
import matplotlib.pyplot as plt
import numpy as np
if __name__ == '__main__':
# 1- 创建画布
plt.figure(figsize=(10,20), dpi=150)
# 2- 准备数据：各个样本点的横纵坐标
x = [i for i in range(1,8)]
y = [np.random.randint(30,40) for i in range(1,8)]
# 3- 绘制图形
plt.plot(x,y)
plt.show()
认识Matplotlib图像结构(了解)
小结
- 什么是matplotlib【了解】
- 是专门用于开发2D(3D)图表的包
- 绘制图像流程【掌握】
- 1.创建画布 – plt.figure(figsize=(20,8), dpi=100)
- 2.绘制图像 – plt.plot(x, y)
- 3.显示图像 – plt.show()
二、常见图形绘制【掌握】
官方案例库
https://matplotlib.org/stable/gallery/index.html
常见图形种类及意义
折线图【掌握】
折线图：以折线的上升或下降来表示统计数量的增减变化的统计图
特点：能够显示数据的变化趋势，反映事物的变化情况。(变化)
api：plt.plot(x, y)
柱形图
柱状图：排列在工作表的列或行中的数据可以绘制到柱状图中。
特点：绘制连离散的数据,能够一眼看出各个数据的大小,比较数据之间的差别。(统计/对比)
api：plt.bar(x, width, align=’center’, **kwargs)
Parameters:
x : 需要传递的数据
width : 柱状图的宽度
align : 每个柱状图的位置对齐方式
{'center', 'edge'}, optional, default: 'center'
**kwargs :
color:选择柱状图的颜色
直方图【掌握】
由一系列高度不等的纵向条纹或线段表示数据分布的情况。 一般用横轴表示数据范围，纵轴表示分布情况。
特点：绘制连续性的数据展示一组或者多组数据的分布状况(统计)
api：matplotlib.pyplot.hist(x, bins=None)
Parameters:
x : 需要传递的数据
bins : 组距
饼图
饼图：用于表示不同分类的占比情况，通过弧度大小来对比各种分类。
特点：分类数据的占比情况(占比)
api：plt.pie(x, labels=,autopct=,colors)
Parameters: x:数量，自动算百分比 labels:每部分名称 autopct:占比显示指定%1.2f%% colors:每部分颜色
散点图【掌握】
散点图：用两组数据构成多个坐标点，考察坐标点的分布,判断两变量之间是否存在某种关联或总结坐标点的分布模式。
特点：判断变量之间是否存在数量关联趋势,展示离群点(分布规律)
api：plt.scatter(x, y)
柱形图绘制
import matplotlib.pyplot as plt
# 中文乱码解决的固定套路
from pylab import mpl
# 设置显示中文字体
mpl.rcParams["font.sans-serif"] = ["SimHei"]
# Mac电脑用下面的代码
# mpl.rcParams["font.sans-serif"] = ['PingFang SC', 'SimHei']
# 有时候，字体更改后，会导致坐标轴中的部分字符无法正常显示，此时需要更改axes.unicode_minus参数：
mpl.rcParams["axes.unicode_minus"] = False
if __name__ == '__main__':
# 1- 创建画布
fig = plt.figure(figsize=(20,10),dpi=100)
# 2- 准备数据
x = ["江西省","广东省","河南省","湖南省"]
y = [10,20,13,16]
# 3- 绘制柱状图
plt.bar(x=x,height=y)
# 4- 其他信息设置
plt.title("各省份学生人数")
plt.xlabel("省份",fontsize=30) # fontsize字体大小
plt.ylabel("人数")
plt.show()
直方图
import matplotlib.pyplot as plt
import numpy as np
# 中文乱码解决的固定套路
from pylab import mpl
# 设置显示中文字体
mpl.rcParams["font.sans-serif"] = ["SimHei"]
# Mac电脑用下面的代码
# mpl.rcParams["font.sans-serif"] = ['PingFang SC', 'SimHei']
# 有时候，字体更改后，会导致坐标轴中的部分字符无法正常显示，此时需要更改axes.unicode_minus参数：
mpl.rcParams["axes.unicode_minus"] = False
if __name__ == '__main__':
# 创建画布
plt.figure(figsize=(10,20),dpi=150)
# 准备数据
data = np.random.randn(500)
# 绘制直方图
"""
数据是如何变成直方图的：
1- 从data数据中获得最小值min、最大值max
2- (max-min)/60每个柱子表示的范围
3- 最后将500个数字按照范围进行划分，每个柱子分到的数字越多柱子的高度就越高
参数解释：
x：要展示的数据
bins：将数据分为多少份
color：柱子的颜色
rwidth：柱子使用的空间是多少。如果不设置，那么柱子间是连在一起的，没有间隙
alpha：颜色的透明度。范围[0,1]，值越小颜色越淡
"""
plt.hist(x=data,bins=60,color="blue",rwidth=0.8,alpha=0.3)
# 其他属性设置
plt.title("直方图展示")
plt.xlabel("分类")
plt.ylabel("占比")
plt.grid()
# 展示图形
plt.show()
饼图
import matplotlib.pyplot as plt
# 设置显示中文字体
from pylab import mpl
mpl.rcParams["font.sans-serif"] = ["SimHei"]
mpl.rcParams["axes.unicode_minus"] = False
if __name__ == '__main__':
# 1- 准备数据
data = [26, 35, 25, 15]
labels = ["湖南省", "广东省", "河南省", "山东省"]
# 2- 绘制图形
# 2.1- 简单版的饼图
# plt.pie(x=data,labels=labels)
# 2.2- 复杂版的饼图
explode = [0, 0.1, 0, 0] # 将 广东省 的扇形突出展示
"""
参数解释:
labels:每个扇形的标签
autopct:对百分比值进行格式化展示
shadow:有阴影,达到3D的效果
startangle:从什么角度开始绘制
explode:将哪些扇形突出展示
"""
plt.pie(x=data,labels=labels,autopct="%.2f%%",shadow=True,startangle=90,explode=explode)
plt.show()
散点图绘制【掌握】
import matplotlib.pyplot as plt
# 设置显示中文字体
from pylab import mpl
mpl.rcParams["font.sans-serif"] = ["SimHei"]
mpl.rcParams["axes.unicode_minus"] = False
if __name__ == '__main__':
# 准备数据
x = [1, 2, 3, 4, 5]
y = [2, 3, 5, 7, 11]
# 绘制图形
plt.scatter(x,y,color="red",alpha=0.7)
plt.title("散点图演示")
plt.xlabel("小时")
plt.ylabel("温度")
plt.show()
三、Matplotlib基础绘图功能【理解】
学习目标
- 掌握给图形添加辅助功能(如：标注、x,y轴名称、标题等)
- 知道图形的保存
- 知道如何多次plot绘制图形
- 知道如何多个坐标系显示图形
折线图的应用场景
呈现公司产品(不同区域)每天活跃用户数
呈现app每天下载数量
呈现产品新功能上线后,用户点击次数随时间的变化
拓展：画各种数学函数图像
- 注意：plt.plot()除了可以画折线图，也可以用于画各种数学函数图像
代码：
import matplotlib.pyplot as plt
import numpy as np
# 中文乱码解决的固定套路
from pylab import mpl
# 设置显示中文字体
mpl.rcParams["font.sans-serif"] = ["SimHei"]
# 有时候，字体更改后，会导致坐标轴中的部分字符无法正常显示，此时需要更改axes.unicode_minus参数：
mpl.rcParams["axes.unicode_minus"] = False
if __name__ == '__main__':
# 1- 准备数据
x = np.linspace(-20,20,1000)
y = np.sin(x)
print(y,type(y))
# 2- 创建画布
plt.figure(figsize=(30,4),dpi=120)
# 3- 绘制图形
plt.plot(x,y)
# 4- 设置其他特性
plt.grid()
# 5- 展示
plt.show()
完善原始折线图 — 给图形添加辅助功能
为了更好地理解所有基础绘图功能，我们通过天气温度变化的绘图来融合所有的基础API使用
需求：画出某城市11点到12点1小时内每分钟的温度变化折线图，温度范围在15度~18度
效果：
准备数据并画出初始折线图
import matplotlib.pyplot as plt
import random
# 画出温度变化图
# 0.准备x, y坐标的数据
x = range(60)
y_shanghai = [random.uniform(15, 18) for i in x]
# 1.创建画布
plt.figure(figsize=(20, 8), dpi=80)
# 2.绘制折线图
plt.plot(x, y_shanghai)
# 3.显示图像
plt.show()
uniform作用：返回一个 随机的浮点数，其值在区间
[a, b]之间（包括a和b）。分布：数值在指定区间内是 均匀分布 的，这意味着每个数值出现的概率是相等的。
添加自定义x,y刻度
plt.xticks(x, **kwargs)
x:要显示的刻度值
plt.yticks(y, **kwargs)
y:要显示的刻度值
# 增加以下两行代码
# 构造x轴刻度标签
x_ticks_label = ["11点{}分".format(i) for i in x]
# 构造y轴刻度
y_ticks = range(40)
# 修改x,y轴坐标的刻度显示
plt.xticks(x[::5], x_ticks_label[::5])
plt.yticks(y_ticks[::5])
如果没有解决过中文问题的话，会显示这个样子：
中文显示问题解决
解决方案一：
下载中文字体（黑体，看准系统版本）
步骤一：下载 SimHei 字体（或者其他的支持中文显示的字体也行）
步骤二：安装字体
linux下：拷贝字体到 usr/share/fonts 下：
sudo cp ~/SimHei.ttf /usr/share/fonts/SimHei.ttfwindows和mac下：双击安装
步骤三：删除~/.matplotlib中的缓存文件
cd ~/.matplotlib rm -r *步骤四：修改配置文件matplotlibrc
vi ~/.matplotlib/matplotlibrc将文件内容修改为：
font.family : sans-serif font.sans-serif : SimHei axes.unicode_minus : False
解决方案二：
在Python脚本中动态设置matplotlibrc,这样也可以避免由于更改配置文件而造成的麻烦，具体代码如下：
from pylab import mpl
# 设置显示中文字体
mpl.rcParams["font.sans-serif"] = ["SimHei"]
# 有时候，字体更改后，会导致坐标轴中的部分字符无法正常显示，此时需要更改axes.unicode_minus参数：
mpl.rcParams["axes.unicode_minus"] = False
添加网格显示
为了更加清楚地观察图形对应的值
plt.grid(True, linestyle='--', alpha=0.5)
添加描述信息
添加x轴、y轴描述信息及标题
通过fontsize参数可以修改图像中字体的大小
# 添加x轴、y轴描述信息及标题
plt.xlabel("时间")
plt.ylabel("温度")
plt.title("中午11点0分到12点之间的温度变化图示", fontsize=20)
图片保存
# 保存图片到指定路径
plt.savefig("test.png")
- 注意：plt.show()会释放figure资源，如果在显示图像之后保存图片将只能保存空图片。
完整代码
import matplotlib.pyplot as plt
import random
# 中文乱码解决的固定套路
from pylab import mpl
# 设置显示中文字体
mpl.rcParams["font.sans-serif"] = ["SimHei"]
# 有时候，字体更改后，会导致坐标轴中的部分字符无法正常显示，此时需要更改axes.unicode_minus参数：
mpl.rcParams["axes.unicode_minus"] = False
if __name__ == '__main__':
# 1- 准备数据
minutes = [i for i in range(60)]
gz_temp = [random.uniform(20,30) for i in range(60)]
# 2- 创建画布
plt.figure(figsize=(30,8), dpi=200)
# 3- 绘制图形
plt.plot(minutes,gz_temp)
# 4- 设置其他信息
# 4.1- 设置刻度尺
x_labels = [f"11点{i}分" for i in range(60)]
# 注意：ticks和labels的个数必须完全一致
plt.xticks(ticks=minutes[::5], labels=x_labels[::5])
# xticks和yticks的个数没有任何关系
y_ticks = [i for i in range(40)]
plt.yticks(ticks=y_ticks[::2])
# 4.2- 设置网格线
plt.grid(visible=True, linestyle="--", alpha=0.4)
# 4.3- 设置标题
plt.title("广州市11点-12点的气温变化",fontsize=20)
plt.xlabel("时间")
plt.ylabel("温度")
# 4.4- 保存图片
# 注意：保存图片需要在show()之前执行
plt.savefig("data/广州市温度.png")
# 5- 展示
plt.show()
在一个坐标系中绘制多个图像
多次plot
需求：再添加一个城市的温度变化
收集到北京当天温度变化情况，温度在1度到3度。怎么去添加另一个在同一坐标系当中的不同图形，其实很简单只需要再次plot即可，但是需要区分线条，如下显示
# 增加北京的温度数据
y_beijing = [random.uniform(1, 3) for i in x]
# 使用多次plot可以画多个折线
plt.plot(x, y_beijing, color='r', linestyle='--')
我们仔细观察，用到了两个新的地方，一个是对于不同的折线展示效果，一个是添加图例。
设置图形风格
整理pdf: https://matplotlib.org/cheatsheets/cheatsheets.pdf
| 颜色字符 | 风格字符 |
|---|---|
| r 红色 | - 实线 |
| g 绿色 | - - 虚线 |
| b 蓝色 | -. 点划线 |
| w 白色 | : 点虚线 |
| c 青色 | ‘ ‘ 留空、空格 |
| m 洋红 | |
| y 黄色 | |
| k 黑色 |
显示图例
- 注意：如果只在plt.plot()中设置label还不能最终显示出图例，还需要通过plt.legend()将图例显示出来。
# 绘制折线图
plt.plot(x, y_shanghai, label="上海")
# 使用多次plot可以画多个折线
plt.plot(x, y_beijing, color='r', linestyle='--', label="北京")
# 显示图例
plt.legend(loc="best")
| Location String | Location Code |
|---|---|
| ‘best’ | 0 |
| ‘upper right’ | 1 |
| ‘upper left’ | 2 |
| ‘lower left’ | 3 |
| ‘lower right’ | 4 |
| ‘right’ | 5 |
| ‘center left’ | 6 |
| ‘center right’ | 7 |
| ‘lower center’ | 8 |
| ‘upper center’ | 9 |
| ‘center’ | 10 |
完整代码
import matplotlib.pyplot as plt
import random
# 中文乱码解决的固定套路
from pylab import mpl
# 设置显示中文字体
mpl.rcParams["font.sans-serif"] = ["SimHei"]
# 有时候，字体更改后，会导致坐标轴中的部分字符无法正常显示，此时需要更改axes.unicode_minus参数：
mpl.rcParams["axes.unicode_minus"] = False
if __name__ == '__main__':
# 1- 准备数据
minutes = [i for i in range(60)]
gz_temp = [random.uniform(20,30) for i in range(60)]
bj_temp = [random.uniform(5,10) for i in range(60)]
other_temp = [random.uniform(1,5) for i in range(60)]
# 2- 创建画布
plt.figure(figsize=(30,8), dpi=200)
# 3- 绘制图形
# 3.1- 广州市的曲线
plt.plot(minutes,gz_temp,label="广州市")
# 3.2- 北京市的曲线
plt.plot(minutes,bj_temp,linestyle="--",color="red", label="北京市")
plt.plot(minutes,other_temp,linestyle=":",color="green", label="某某市")
# 4- 设置其他信息
# 4.1- 设置刻度尺
x_labels = [f"11点{i}分" for i in range(60)]
# 注意：ticks和labels的个数必须完全一致
plt.xticks(ticks=minutes[::5], labels=x_labels[::5])
# xticks和yticks的个数没有任何关系
y_ticks = [i for i in range(40)]
plt.yticks(ticks=y_ticks[::2])
# 4.2- 设置网格线
plt.grid(visible=True, linestyle="--", alpha=0.4)
# 4.3- 设置标题
plt.title("广州市11点-12点的气温变化",fontsize=20)
plt.xlabel("时间")
plt.ylabel("温度")
# 4.4- 展示图例
# plt.legend(loc="best") # best：Matplotlib自动帮我们选择最合适的地方
plt.legend(loc="upper left")
# 4.5- 保存图片
# 注意：保存图片需要在show()之前执行
plt.savefig("data/广州市温度.png")
# 5- 展示
plt.show()
多个坐标系显示— plt.subplots(面向对象的画图方法)
如果我们想要将上海和北京的天气图显示在同一个图的不同坐标系当中，效果如下：
可以通过subplots函数实现(旧的版本中有subplot，使用起来不方便)，推荐subplots函数
matplotlib.pyplot.subplots(nrows=1, ncols=1, **fig_kw) 创建一个带有多个axes(坐标系/绘图区)的图
Parameters: nrows, ncols : 设置有几行几列坐标系 int, optional, default: 1, Number of rows/columns of the subplot grid. Returns: fig : 图对象 axes : 返回相应数量的坐标系 设置标题等方法不同： set_xticks set_yticks set_xlabel set_ylabel关于axes子坐标系的更多方法：参考https://matplotlib.org/api/axes_api.html#matplotlib.axes.Axes
注意：**plt.函数名()相当于面向过程的画图方法，axes.set_方法名()**相当于面向对象的画图方法。
import matplotlib.pyplot as plt
import random
from matplotlib.axes import Axes
# 中文乱码解决的固定套路
from pylab import mpl
# 设置显示中文字体
mpl.rcParams["font.sans-serif"] = ["SimHei"]
# 有时候，字体更改后，会导致坐标轴中的部分字符无法正常显示，此时需要更改axes.unicode_minus参数：
mpl.rcParams["axes.unicode_minus"] = False
if __name__ == '__main__':
# 1- 准备数据
minutes = [i for i in range(60)]
gz_temp = [random.uniform(20,30) for i in range(60)]
bj_temp = [random.uniform(5,10) for i in range(60)]
# 2- 创建画布
"""
将figsize=(30,8)画布分为1行2列。也就是有2个坐标系
"""
figs,axes = plt.subplots(nrows=1,ncols=2,figsize=(30,8), dpi=200)
# 导包的快捷键： alt+回车
axes_0:Axes = axes[0]
axes_1:Axes = axes[1]
# 3- 绘制图形
axes_0.plot(minutes,gz_temp,color="green")
axes_1.plot(minutes,bj_temp,color="red")
# 4- 设置其他信息
# 4.1- 设置刻度尺
x_labels = [f"11点{i}分" for i in range(60)]
y_ticks = [i for i in range(40)]
# 第一个图形的刻度尺
axes_0.set_xticks(ticks=minutes[::5], labels=x_labels[::5])
axes_0.set_yticks(ticks=y_ticks[::2])
# 第二个图形的刻度尺
axes_1.set_xticks(ticks=minutes[::5], labels=x_labels[::5])
axes_1.set_yticks(ticks=y_ticks[::2])
# 4.2- 设置网格线
axes_0.grid(visible=True, linestyle="--", alpha=0.4)
axes_1.grid(visible=True, linestyle=":", alpha=0.4)
# 4.3- 设置标题
axes_0.set_title("广州市11点-12点的气温变化",fontsize=20)
axes_0.set_xlabel("时间")
axes_0.set_ylabel("温度")
axes_1.set_title("北京市11点-12点的气温变化", fontsize=20)
axes_1.set_xlabel("时间")
axes_1.set_ylabel("温度")
# 4.4- 保存图片
plt.savefig("data/各个市温度.png")
# 5- 展示
plt.show()
版权声明
本文为 程序员青阳 原创文章，遵循 CC BY-NC-SA 4.0 版权协议，转载请附上原文链接及本声明。