Mean square displacement of a 1d random walk in python

Question

I\'d like to calculate the mean square displacement at all times of a 1d trajectory. I\'m a novice programmer, so I\'ve attempted it from a simulated random walk.

<