Plasma Flow in Tokamak

import matplotlib.pyplot as plt
import numpy as np

# Toroidal plasma flow (tokamak-like)
n = 6
theta = np.linspace(0, 2*np.pi, n)
phi = np.linspace(0, 2*np.pi, n)
r_minor = np.linspace(0.3, 0.8, 3)

# Create toroidal grid
THETA, PHI, R = np.meshgrid(theta, phi, r_minor)
R_major = 1.5

X = (R_major + R * np.cos(THETA)) * np.cos(PHI)
Y = (R_major + R * np.cos(THETA)) * np.sin(PHI)
Z = R * np.sin(THETA)

# Toroidal + poloidal flow
U = -np.sin(PHI) * 0.5 - R * np.sin(THETA) * np.cos(PHI) * 0.3
V = np.cos(PHI) * 0.5 - R * np.sin(THETA) * np.sin(PHI) * 0.3
W = R * np.cos(THETA) * 0.3

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

ax.quiver(X, Y, Z, U, V, W, length=0.2, normalize=True,
          color='#22d3ee', alpha=0.85, arrow_length_ratio=0.3)

ax.set_xlim(-3, 3)
ax.set_ylim(-3, 3)
ax.set_zlim(-1.5, 1.5)

ax.set_xlabel('X', color='white', fontsize=10)
ax.set_ylabel('Y', color='white', fontsize=10)
ax.set_zlabel('Z', color='white', fontsize=10)
ax.set_title('Plasma Flow in Tokamak', 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=25, azim=45)
plt.tight_layout()
plt.show()