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

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()

文章链接

 您阅读本篇文章共花了：