30

I want to draw x=0 and y=0 axis in my contour plot, using a white color. If that is too cumbersome, I would like to have a white dot denoting where the origin is.

My contour plot looks as follows and the code to create it is given below.

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

Hooked
  • 84,485
  • 43
  • 192
  • 261
nos
  • 19,875
  • 27
  • 98
  • 134

2 Answers2

90

There are a number of options (E.g. centered spines), but in your case, it's probably simplest to just use axhline and 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

Joe Kington
  • 275,208
  • 71
  • 604
  • 463
  • 3
    `spine.set_position('zero')` is the way to go, why do you recommend drawing lines manually instead? – azrdev Aug 07 '18 at 01:07
  • Is it possible to use what the above comment suggests without having the axis object? – scribe Feb 05 '23 at 21:21
1

Can't you just overlay a straight line? 

plt.plot([0,0],[-4,4],lw=3,'w')
ev-br
  • 24,968
  • 9
  • 65
  • 78