Создание трехмерных контурных графиков

Я пытался получить "примерное" изображение (сгенерированное с помощью трехмерного графического расчета) с помощью Python в течение нескольких дней, но продолжаю сталкиваться с проблемами, получая сегменты графика, отличные от пика в середине, для отображения в масштабе .

Я использую этот код:

import numpy as np
from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111, projection = "3d")
X, Y = np.mgrid[-1:1:30j, -1:1:30j]
Z = (2*X*Y) + (1/np.sqrt(X**2+Y**2))
ax.plot_surface(X, Y, Z, cmap = "autumn_r", lw=0.5, rstride=1, cstride=1)
ax.contour(X, Y, Z, 10, lw=3, cmap = "autumn_r", linestyles = "solid", offset=-1)
ax.contour(X, Y, Z, 10, lw=3, colors = "k", linestyles = "solid")
plt.show()

Что дает этот график.

Он близок, но должен быть похож на этот. Когда я опускаю 30j в попытках сбить его и надеюсь, что блики по бокам более выражены, это избавляется от всего пика. Я пытаюсь добраться до этого.