#-------------------------------------------------------------------------------
# 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()