Šroubovice

importujeme knihovny numpy (na numerické výpočty) a matplotlib (na kreslení grafů)

import numpy as np
import matplotlib.pyplot as plt

Zadáme parametry pro rovnoměrný pohyb po kružnici:
$r$ je poloměr kruhové trajektorie
$T$ je perioda pohybu (tj. doba za kterou hmotný bod oběhne celou kružnici)
úhlová frekvence pohybu bude tedy $\omega = \frac{2\pi}{T}$

r = 1 #poloměr trajektorie 1 m
T = 1 #perioda pohybu 1 s
omega = 2 *np.pi/T #=uhlová frekvence

Zvolíme rychlost v_z, se kterou se hmotný bod pohybuje kolmo na rovinu $xy$

v_z = T/10 #hmotný bod se pohybuje kolmo na rovinu xy rychlostí 10-krát menší než je oběhová rychlost  

Vytvoříme pole t obsahující časy, v kterých chceme vypočítat souřadnice hmotného bodu

t = np.linspace(0,5*T,5000) #rozdělíme periodu na 1000 částí

Vypočítáme souřadnice hmotného bodu v jednotlivých časech
$$x = r\, \text{cos}(\omega t)$$ $$y = r\, \text{sin}(\omega t)$$ $$z = v_z t$$ Tyto souřadnice budou v polích x, ya z

x=r*np.cos(omega*t) #x-ová souřadnice
y=r*np.sin(omega*t) #y-ová souřadnice
z=v_z*t #z-ová souřadnice

Nyní můžeme nakreslit graf trajektorie

ax=plt.figure().add_subplot(projection='3d') #projekce 3D grafu
ax.plot(x,y,z) #vytvoření grafu
ax.set_xlabel("x") #označení osy x
ax.set_ylabel("y") #označení osy y
ax.set_zlabel("z") #označení osy z
plt.show() #vykreslení grafu