绘制Matplotlib等高线图的轴线或原点

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

有许多选项(例如居中的刺),但在你的情况下,它可能是最简单的使用axhline和axvline.

例如

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