circular polarized

1import numpy as np
2import matplotlib.pyplot as plt
3from mpl_toolkits.mplot3d import Axes3D
4
5n = 1000
6fig = plt.figure()
7ax = fig.add_subplot(111, projection='3d')
8
9# Plot a helix along the x-axis
10theta_max = 8 * np.pi
11theta = np.linspace(0, theta_max, n)
12x = theta
13z =  np.sin(theta)
14y =  np.cos(theta)
15ax.plot(x, y, z, 'b', lw=2)
16
17# An line through the centre of the helix
18ax.plot((-theta_max*0.2, theta_max * 1.2), (0,0), (0,0), color='k', lw=2)
19# sin/cos components of the helix (e.g. electric and magnetic field
20# components of a circularly-polarized electromagnetic wave
21ax.plot(x, y, 0, color='r', lw=1, alpha=0.5)
22ax.plot(x, [0]*n, z, color='m', lw=1, alpha=0.5)
23
24# Remove axis planes, ticks and labels
25ax.set_axis_off()
26plt.show()
27