python matplotlib contour画等高线图

以下，我将一些常用的功能补充全代码，如下：

#导入模块
import numpy as np
import matplotlib.pyplot as plt

#建立步长为0.01，即每隔0.01取一个点
step = 0.01
x = np.arange(-10,10,step)
y = np.arange(-10,10,step)
#也可以用x = np.linspace(-10,10,100)表示从-10到10，分100份

#将原始数据变成网格数据形式
X,Y = np.meshgrid(x,y)
#写入函数，z是大写，这里我让中间的0是最大，加了一个负号
Z = -(X**2+Y**2)
#填充颜色，f即filled,6表示将三色分成三层，cmap那儿是放置颜色格式，hot表示热温图（红黄渐变）
#更多颜色图参考：https://blog.csdn.net/mr_cat123/article/details/80709099
#颜色集，6层颜色，默认的情况不用写颜色层数,
cset = plt.contourf(X,Y,Z,6,cmap=plt.cm.hot) 
#or cmap='hot'

#画出8条线，并将颜色设置为黑色
contour = plt.contour(X,Y,Z,8,colors='k')
#等高线上标明z（即高度）的值，字体大小是10，颜色分别是黑色和红色
plt.clabel(contour,fontsize=10,colors='k')
#去掉坐标轴刻度
#plt.xticks(())  
#plt.yticks(())  
#设置颜色条，（显示在图片右边）
plt.colorbar(cset)
#显示
plt.show()

————————————————
版权声明：本文为CSDN博主「Mr-Cat伍可猫」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/Mr_Cat123/article/details/80677525