Mexican Hat Wavelet (Ricker)

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import LinearSegmentedColormap

colors = ['#0a0a0f', '#052e16', '#22c55e', '#86efac']
cmap = LinearSegmentedColormap.from_list('matrix', colors, N=256)

X = np.linspace(-5, 5, 100)
Y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(X, Y)
R2 = X**2 + Y**2
sigma = 1
Z = (1 - R2/sigma**2) * np.exp(-R2/(2*sigma**2))  # Ricker/Mexican Hat wavelet

fig = plt.figure(figsize=(10, 8), facecolor='#0a0a0f')
ax = fig.add_subplot(111, projection='3d', facecolor='#0a0a0f')

surf = ax.plot_surface(X, Y, Z, cmap=cmap, linewidth=0, antialiased=True)

ax.set_xlabel('X', color='white', fontsize=10)
ax.set_ylabel('Y', color='white', fontsize=10)
ax.set_zlabel('ψ(x,y)', color='white', fontsize=10)
ax.set_title('Mexican Hat Wavelet (Ricker)', color='white', fontsize=14, fontweight='bold', pad=20)

ax.tick_params(colors='#64748b', labelsize=8)
ax.xaxis.pane.fill = False
ax.yaxis.pane.fill = False
ax.zaxis.pane.fill = False
ax.xaxis.pane.set_edgecolor('#1e293b')
ax.yaxis.pane.set_edgecolor('#1e293b')
ax.zaxis.pane.set_edgecolor('#1e293b')

ax.view_init(elev=30, azim=45)
plt.tight_layout()
plt.show()