python中怎么在等高线图中添加参考线？

#导入模块

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是大写

Z = X**2+Y**2

#填充颜色，f即filled

plt.contourf(X,Y,Z)

#画等高线

contour = plt.contour(X,Y,Z)

#平行于坐标系的参考线

plt.axhline(y=2.5,c='red',ls='--',lw=3)

plt.axvline(x=2,c='green',ls='--',lw=3)

#等高线上标明z（即高度）的值，默认是保留3个小数，可以如下保留四位

plt.clabel(contour,fontsize=10,colors=('k','r'),fmt='%.2f')

plt.show()

差不多就这样吧，有问题的话再留言吧