py525 hw2 - 2

import numpy
import pylab

nsteps = numpy.linspace(100,1000,10)
nstores = numpy.zeros(nsteps.size)
for i in range (0, nsteps.size):
    rvalues = numpy.zeros(40)
    for j in range (0,rvalues.size):
        randomx = numpy.zeros(10**6)
        randomy = numpy.zeros(10**6)
        for k in range(1,randomx.size):
            rvalue = numpy.random.randint(1,4)
            if rvalue == 1:
                randomx[k] = randomx[k-1]-1
                randomy[k] = randomy[k-1]
            if rvalue == 2:
                randomx[k] = randomx[k-1]+1
                randomy[k] = randomy[k-1]
            if rvalue == 3:
                randomx[k] = randomx[k-1]
                randomy[k] = randomy[k-1]-1
            if rvalue == 4:
                randomx[k] = randomx[k-1]
                randomy[k] = randomy[k-1]+1
        rvalues[j] = numpy.sqrt((randomx[-1])**2 (randomy[-1])**2)
    nstores[i] = (numpy.sum(rvalues))/40.