There are several classes designed for working with particles in a plasma.
slactrac.Scatter.theta_rms
Elegant