matplotlib绘图示例

# 导入必要的库
import matplotlib.pyplot as plt  # 导入matplotlib的pyplot模块,用于绘图
import numpy as np  # 导入numpy库,用于数值计算

# 创建一个从-4到4的整数序列,步长为1
points = np.arange(-4, 4, 1)

# 使用meshgrid函数创建二维网格坐标
# xm和ym分别是网格的x坐标和y坐标矩阵
xm, ym = np.meshgrid(points, points)

# 计算每个网格点到原点的欧几里得距离
zm = np.sqrt(xm**2 + ym**2)

# 创建一个3x3英寸的图形
plt.figure(figsize=(3, 3))

# 使用imshow函数显示距离矩阵zm,使用粉色调色板
plt.imshow(zm, cmap="pink")

# 添加颜色条,显示颜色与值的对应关系
plt.colorbar()

# 将图形保存为b.png文件
plt.savefig("b.png")

# 关闭当前图形,释放资源
plt.close()
posted @ 2025-05-20 23:35  卓能文  阅读(21)  评论(0)    收藏  举报