python - 为 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

有许多选项(例如 centered spines )，但在您的情况下，使用 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()