Pyplot 是 Matplotlib 的子库,提供了和 MATLAB 类似的绘图 API。
Pyplot 是常用的绘图模块,能很方便让用户绘制 2D 图表。
Pyplot 包含一系列绘图函数的相关函数,每个函数会对当前的图像进行一些修改,例如:给图像加上标记,生新的图像,在图像中产生新的绘图区域等等。
1. 导入 pyplot 模块:
使用的时候,我们可以使用 import 导入 pyplot 库,并设置一个别名 plt:
import matplotlib.pyplot as plt
以下是一些常用的 pyplot 函数:
plot()
:用于绘制线图和散点图
scatter()
:用于绘制散点图
bar()
:用于绘制垂直条形图和水平条形图
hist()
:用于绘制直方图
pie()
:用于绘制饼图
imshow()
:用于绘制图像
subplots()
:用于创建子图
2. 创建简单的折线图:
# 导入Matplotlib库和NumPy库
import matplotlib.pyplot as plt
import numpy as np# 创建包含点的一维NumPy数组
xpoints = np.array([0, 6]) # x坐标点集合
ypoints = np.array([0, 100]) # y坐标点集合# 使用plot函数绘制折线图
plt.plot(xpoints, ypoints) # 绘制从(0, 0)到(6, 100)的直线# 显示绘制的图形
plt.show()
3. 自定义图表:
# 导入Matplotlib库和NumPy库
import matplotlib.pyplot as plt
import numpy as np# 定义数据点的 x 和 y 值
x = [1, 2, 3, 4] # x 值
y = [10, 15, 13, 18] # y 值# 使用 plot 函数绘制线条图,并设置线条的样式,颜色,标记和标签
plt.plot(x, y, color='red', linestyle='--', marker='o', label='Data Points')# 设置 x 轴标签
plt.xlabel('X-axis')# 设置 y 轴标签
plt.ylabel('Y-axis')# 设置图表标题
plt.title('Simple Line Plot')# 显示图例
plt.legend()# 显示网格线
plt.grid(True)# 显示绘制的图形
plt.show()
4. 绘制多个图形:
# 导入Matplotlib库和NumPy库
import matplotlib.pyplot as plt
import numpy as np# 创建第一个子图(2行1列,第一个子图)
plt.subplot(2, 1, 1)
plt.plot([1, 2, 3, 4]) # 在第一个子图中绘制折线图,x轴默认为[0, 1, 2, 3]# 创建第二个子图(2行1列,第二个子图)
plt.subplot(2, 1, 2)
plt.plot([4, 3, 2, 1]) # 在第二个子图中绘制折线图,x轴默认为[0, 1, 2, 3]# 显示绘制的图形
plt.show()
5. 绘制散点图:
# 导入Matplotlib库和NumPy库
import matplotlib.pyplot as plt
import numpy as np# 定义数据点的 x 和 y 值,以及散点的大小
x = [1, 2, 3, 4] # x 值
y = [10, 15, 13, 18] # y 值
sizes = [20, 50, 80, 200] # 散点大小# 使用scatter函数绘制散点图,并设置散点的大小、颜色和透明度
plt.scatter(x, y, s=sizes, c='r', alpha=0.5) # 设置散点的大小、颜色为红色、透明度为0.5# 显示绘制的散点图
plt.show()
6. 绘制条形图:
# 导入Matplotlib库和NumPy库
import matplotlib.pyplot as plt
import numpy as np# 定义条形图的 x 和 y 值
x = ['A', 'B', 'C', 'D'] # 类别
y = [10, 20, 15, 25] # 数值# 使用bar函数绘制条形图,并设置颜色为蓝色
plt.bar(x, y, color='blue')# 设置 x 轴标签
plt.xlabel('Categories')# 设置 y 轴标签
plt.ylabel('Values')# 设置图表标题
plt.title('Bar Chart')# 显示绘制的图形
plt.show()