为Matplotlib等高线p绘制轴线或原点

Question

我想在等高线图中用白色绘制x=0和y=0轴。如果太麻烦的话，我想用一个白点来表示原点在哪里。

我的等高线图如下所示，创建它的代码如下所示。

xvec = linspace(-5.,5.,100)                               
X,Y = meshgrid(xvec, xvec)                                
fig = plt.figure(figsize=(6, 4))                      
contourf(X, Y, W,100)                             
plt.colorbar()                                    

Answer 1

你就不能画一条直线吗？

plt.plot([0,0],[-4,4],lw=3,'w')

Answer 2

有很多选项（例如centered spines），但是在您的例子中，使用^{}和^{}可能是最简单的。

例如

import numpy as np
import matplotlib.pyplot as plt

xvec = np.linspace(-5.,5.,100)                               
x,y = np.meshgrid(xvec, xvec)
z = -np.hypot(x, y)                                

plt.contourf(x, y, z, 100)                             
plt.colorbar() 

plt.axhline(0, color='white')
plt.axvline(0, color='white')

plt.show()