Vortex Flow Pattern

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

# Polar sampling for vortex
n_radii = 18
n_angles = 50
radii = np.linspace(0.2, 2.5, n_radii)
angles = np.linspace(0, 2*np.pi, n_angles, endpoint=False)[..., np.newaxis]

x = (radii * np.cos(angles)).flatten()
y = (radii * np.sin(angles)).flatten()
r = np.sqrt(x**2 + y**2)

# Rankine vortex velocity
r_core = 0.8
v_theta = np.where(r < r_core, r/r_core, r_core/r)
z = v_theta

# Custom colormap
colors = ['#0a0a0f', '#431407', '#F54927', '#F5B027', '#fef08a']
cmap = LinearSegmentedColormap.from_list('orange_red', colors, N=256)

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

surf = ax.plot_trisurf(x, y, z, cmap=cmap, alpha=0.9, linewidth=0.1, edgecolor='#ffffff08')

ax.set_xlabel('X', fontsize=10, color='#94a3b8', labelpad=10)
ax.set_ylabel('Y', fontsize=10, color='#94a3b8', labelpad=10)
ax.set_zlabel('Velocity', fontsize=10, color='#94a3b8', labelpad=10)
ax.set_title("Vortex Flow Pattern", fontsize=14, color='white', 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()