mplot3d_demo.py

#! /usr/bin/python3
# ==================================================================
# 3D plot
#
# Found on: https://matplotlib.org/examples/index.html
#
# From: matplotlib.org/examples/mplot3d/lines3d_demo.html
# 
# ==================================================================

import matplotlib as mpl
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.pyplot as plt

mpl.rcParams['legend.fontsize'] = 10

fig = plt.figure()

ax = fig.gca(projection='3d')
theta = np.linspace(-4 * np.pi, 4 *np.pi,100)

z = np.linspace(-2,2,100)
r = z**2 + 1
x = r * np.sin(theta)
y = r * np.cos(theta)

ax.plot(x,y,z,label='parametric curve')
ax.legend()

plt.show()