import matplotlib as mpl
import matplotlib.animation as mplanim
import matplotlib.pyplot as plt
import numpy as np

mpl.rcParams['figure.dpi'] = 200

fig = plt.figure()
ax = fig.add_subplot(projection='3d')

# Prepare arrays x, y, z
#theta = np.linspace(-2 * np.pi, 2 * np.pi, 200)
theta = np.linspace(-0 * np.pi, 2 * np.pi, 200)
z = theta  # (theta/2*np.pi)*36
r = 1
x = r * np.cos(theta)
y = r * np.sin(theta)

ax.plot(x, y, z, label=r'$ x,y,z = cos(\theta), sin(\theta), \theta_{radians} $',)
ax.legend()
ax.set_xlabel(r'$ x = cos(\theta) $')
ax.set_ylabel(r'$ y = sin(\theta) $')
ax.set_zlabel(r'$ z = \theta $')
ax.set_proj_type('ortho')
ax.set_proj_type('persp', focal_length=0.2)

ax.plot(x, y, zs=0, zdir='z', label='curve in (x, y)', alpha=0.5)
ax.plot(x, y, zs=2*np.pi, zdir='z', label='curve in (x, y)', alpha=0.5)

ax.view_init(elev=10., azim=-35, roll=10, vertical_axis='x')
plt.show()

def init():
    #ax.scatter(xx, yy, zz, marker='o', s=20, c="goldenrod", alpha=0.6)
    return fig,

def animate(i):
    ax.view_init(elev=10., azim=i, roll=0, vertical_axis='x')
    return fig,

# Animate
anim = mplanim.FuncAnimation(fig, animate, init_func=init,
                               frames=360, interval=20, blit=True)
# Save an .ogv video
codec, ext, mimetype = 'libtheora', '.ogv', 'video/ogg'



codec, ext, mimetype = 'webp', '.webp', 'image/webp'
filename = f'basic_animation.{ext}'
writer = mplanim.FFMpegWriter(fps=30, codec=codec, extra_args=['-loop', '0'])
anim.save(filename, writer=writer)

from IPython.display import Image
# Image(filename, embed=True)  #  ~ 11 MB as bas64 data uri data
Image(filename)