I need the function to be able to walk in all four directions with equal probabbility and return the distance away from the starting point. The function below only takes steps forwards and backwards?
def main(): numWalks, numSteps = getInputs() averageSteps = takeWalks(numWalks, numSteps) printExpectedDistance(averageSteps) def getInputs(): numWalks = input("How many random walks to take? ") numSteps = input("How many steps for each walk? ") return numWalks, numSteps def takeWalks(numWalks, numSteps): totalSteps = 0 for walk in range(numWalks): stepsAway = takeAWalk(numSteps) totalSteps = totalSteps + stepsAway return float(totalSteps) / numWalks def printExpectedDistance(averageSteps): print "The expected number of steps away from the " print "start point is", averageSteps def takeAWalk(numSteps): from random import random stepsForwardOfStart = 0 for step in range(numSteps): if random() < 0.5: stepsForwardOfStart = stepsForwardOfStart - 1 else: stepsForwardOfStart = stepsForwardOfStart + 1 return abs(stepsForwardOfStart) main()