Python学习笔记——绘图设置（二）填色图与等值线图

最新推荐文章于 2024-04-20 22:01:52 发布

风萧萧兮_x6

最新推荐文章于 2024-04-20 22:01:52 发布

阅读量4k

点赞数

文章标签： python 学习 matplotlib

本文链接：https://blog.csdn.net/qq_52304378/article/details/126980749

版权

import numpy as np
import matplotlib.pyplot as plt
delta = 0.025
x = np.arange(-3., 3., delta)  # 创建x,y列表 注意x、y的shape为（240，）
y = np.arange(-2., 2., delta)
# print(x.shape)
X, Y = np.meshgrid(x, y)  # 将x,y转化为X,Y,形状变为（160,240）==> 为得到数据Z
# print(X.shape)
# print(x)
# print('----------------')
# print(X)
Z1 = np.exp(-X**2 - Y**2)
Z2 = np.exp(-(X-1)**2 - (Y-1)**2)
Z = (Z1 - Z2) * 2
fg, ax = plt.subplots()  # 添加画布对象
lv1 = np.arange(-2, 2.1, 0.4)  # 生成等值线序列
qs = ax.contourf(x, y, Z, levels=lv1, cmap='RdBu_r')  # 绘制填色图  camp可以创建或者使用预设
cs = ax.contour(x, y, Z, levels=lv1, colors='k')  # 绘制等值线
ax.clabel(cs, fontsize=8, fmt='% 3.1f')    # 在cs上标注等值线
fg.colorbar(qs, shrink=0.8)    # 在画布上添加以qs中绘图数据为参考的色标
plt.savefig("D:\\fang\\shaded.pdf") 
fg.show()