Whitney Umbrella Surface

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

# Whitney umbrella parametrization
u = np.linspace(-1.5, 1.5, 80)
v = np.linspace(-1.5, 1.5, 80)
u, v = np.meshgrid(u, v)

x = u * v
y = u
z = v**2

# Custom colormap
colors = ['#ffffff', '#dbeafe', '#276CF5', '#27D3F5', '#a5f3fc']
cmap = LinearSegmentedColormap.from_list('blue_cyan', colors, N=256)

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

surf = ax.plot_surface(x, y, z, cmap=cmap, alpha=0.85, linewidth=0.1, edgecolor='#33333318')

ax.set_xlabel('X', fontsize=10, color='#374151', labelpad=10)
ax.set_ylabel('Y', fontsize=10, color='#374151', labelpad=10)
ax.set_zlabel('Z', fontsize=10, color='#374151', labelpad=10)
ax.set_title("Whitney Umbrella Surface", fontsize=14, color='#1f2937', fontweight='bold', pad=20)

ax.tick_params(colors='#6b7280', labelsize=8)
ax.xaxis.pane.fill = False
ax.yaxis.pane.fill = False
ax.zaxis.pane.fill = False
ax.xaxis.pane.set_edgecolor('#e5e7eb')
ax.yaxis.pane.set_edgecolor('#e5e7eb')
ax.zaxis.pane.set_edgecolor('#e5e7eb')

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