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

时间:2012-03-07 21:30:56

标签: python matplotlib contour

我想在我的轮廓图中使用白色绘制x=0y=0轴。如果这太麻烦了,我想要一个白点来表示原点。

我的轮廓图如下所示,下面给出了创建它的代码。

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

enter image description here

2 个答案:

答案 0 :(得分:68)

有很多选项(例如centered spines),但在您的情况下,使用axhlineaxvline可能最简单。

E.g。

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

enter image description here

答案 1 :(得分:1)

你不能只覆盖一条直线吗?

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