概述
在上一节,我们介绍了Python的pandas模块,包括:Series、DataFrame、数据读取和写入等内容。在这一节,我们将介绍Python的matplotlib模块。matplotlib模块是一个Python的2D绘图库,可以实现各种类型的图形绘制,包括:线图、柱状图、饼图、散点图等。matplotlib支持各种格式的输出,比如:PNG、SVG、PDF等,可以很方便地将图形输出到文件或在屏幕上显示。
在Python中使用matplotlib模块,需要先安装matplotlib库。可以通过pip命令进行安装:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple matplotlib。安装完成后,就可以在Python脚本中导入matplotlib模块,并使用其函数和方法了。
线图
在下面的示例代码中,我们首先引入了Matplotlib的pyplot模块,并定义了x、y两个列表。然后,使用plt.plot(x, y)函数绘制了一条线。最后,通过plt.show()函数显示图形。
import matplotlib.pyplot as plt
x = [1, 3, 5, 7, 9]
y = [1, 9, 25, 49, 81]
plt.plot(x, y)
plt.show()
执行上面的示例代码后,其输出如下。
除了基本的绘图函数,Matplotlib还提供了许多参数设置功能,可以自定义图形的各种属性,比如:颜色、线型、刻度、标签等。下面的示例代码绘制了一条黄色的线,并设置了X轴和Y轴的标签。
import matplotlib.pyplot as plt
x = [1, 3, 5, 7, 9]
y = [1, 9, 25, 49, 81]
plt.plot(x, y, color = 'yellow')
plt.xlabel('X Axis')
plt.ylabel('Y Axis')
plt.show()
执行上面的示例代码后,其输出如下。
柱状图
在下面的示例代码中,我们使用plt.bar()函数绘制了一个简单的柱状图,并传入了标签和相应的数值,图形中的每个柱子都对应一个标签和一个数值。
import matplotlib.pyplot as plt
labels = ['C', 'S', 'D', 'N']
values = [36, 50, 22, 65]
plt.bar(labels, values)
plt.show()
执行上面的示例代码后,其输出如下。
饼图
在下面的示例代码中,我们使用plt.pie()函数绘制了一个简单的饼图,并传入了标签和相应的数值。plt.pie()函数有如下几个参数,分别进行介绍。
sizes:表示每个部分的大小,它将自动根据这些大小生成饼图。
labels:表示每个部分的标签。
autopct:用于显示每个部分的百分比,格式由参数值决定。在本例中,它显示的是百分比的小数形式,保留两位小数。
import matplotlib.pyplot as plt
labels = ['C', 'S', 'D', 'N']
values = [36, 50, 22, 65]
# 保留两位小数
plt.pie(values, labels = labels, autopct = '%1.2f%%')
# 确保饼图是圆的(即x轴和y轴的比例相同)
plt.axis('equal')
plt.show()
执行上面的示例代码后,其输出如下。
散点图
在下面的示例代码中,我们使用plt.scatter(x, y)函数创建了散点图。该函数接受两个参数:x和y坐标列表。然后,我们使用plt.title()、plt.xlabel()、plt.ylabel()函数为散点图添加了标题和轴标签。
import matplotlib.pyplot as plt
import numpy as np
# 生成一些随机数据
x = np.random.rand(30)
y = np.random.rand(30)
# 创建散点图
plt.scatter(x, y)
# 添加标题和轴标签
plt.title('Scatter Plot Demo')
plt.xlabel('X-Axis')
plt.ylabel('Y-Axis')
plt.show()
执行上面的示例代码后,其输出如下。
添加文本及矩形
对于定制化需求,Matplotlib模块提供了大量的参数和API供用户自定义图形。比如:可以设置图形的标题、标签、颜色、线型、刻度等属性,也可以添加文本、散点、矩形等图形元素。
在下面的示例代码中,我们首先使用plt.text函数添加了文本。然后,使用plt.Rectangle函数来创建一个矩形对象,并使用ax.add_patch函数将它添加到图形上。矩形的位置和大小由它的四个参数决定:左下角的x和y坐标、矩形的宽度和高度。另外,我们还可以设置矩形的颜色和线宽。最后,我们设置x和y轴的限制,以便全部的矩形都能被显示出来。
import matplotlib.pyplot as plt
x = [1, 3, 5, 7, 9]
y = [1, 9, 25, 49, 81]
plt.plot(x, y)
# 添加文本
plt.text(5, 36, 'CSDN', fontsize = 12)
# 添加矩形
fig, ax = plt.subplots()
rect = plt.Rectangle((3, 3), 6, 9, fill = True, edgecolor = 'red', linewidth = 2)
ax.add_patch(rect)
# 设置x和y轴的限制,以便显示全部的矩形
ax.set_xlim(0, 15)
ax.set_ylim(0, 15)
plt.show()
执行上面的示例代码后,其输出如下。