Нарисовать оси оси или начало координат для контура контура Matplotlib

Я хочу нарисовать ось 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()

enter image description here

Ответы

Ответ 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()

enter image description here

Ответ 2

Не можете ли вы просто наложить прямую линию?

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