2012-03-07 18 views
16

Quiero dibujar los ejes x=0 y y=0 en mi gráfica de contorno, usando un color blanco. Si eso es demasiado engorroso, me gustaría tener un punto blanco que indique dónde está el origen.Dibujar líneas de ejes o el origen de la gráfica de contorno de Matplotlib

Mi diagrama de contorno se ve de la siguiente manera y el código para crearlo se muestra a continuación.

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

Respuesta

54

Hay una serie de opciones (por ejemplo centered spines), pero en su caso, probablemente es más simple de usar sólo axhline y axvline.

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

no puedes superponer una línea recta?

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