Random Walk and pyplot simple demo

In [2]:
#-------------------------------------------------------------------------------
# Name:        RandomWalk.py
# Purpose:     Simple demo of Random Walk and pyplot, Python 2.7 version.
#
# Author:      Jennifer Yoon
# Created:     07/13/2016
# URL for downloading:
# https://notebooks.azure.com/JennYoon/libraries/public2017
# https://notebooks.azure.com/run/JennYoon/public2017?dest=/notebooks/RandomWalk.ipynb
#-------------------------------------------------------------------------------
%matplotlib inline
import random  # Python random.py module, NOT NumPy.
import matplotlib.pyplot as plt

position = 0
walk = [position]
steps = 500
for i in xrange(steps):
    step = 1 if random.randint(0, 1) else -1
    position += step
    walk.append(position)
##    print "i:", i, "step:", step, "walk:", walk
plt.plot(walk)
plt.xlabel('steps')
plt.ylabel('position')
plt.show()
In [ ]:
 
In [ ]: