def getFinalLocs(numSteps, numTrials, dClass):
locs = []
d = dClass()
for t in range(numTrials):
f = Field()
f.addDrunk(d, Location(0, 0))
for s in range(numSteps):
f.moveDrunk(d)
locs.append(f.getLoc(d))
return locs
def traceWalk(drunkKinds, numSteps):
styleChoice = styleIterator(('k+', 'r^', 'mo'))
f = Field()
for dClass in drunkKinds:
d = dClass()
f.addDrunk(d, Location(0, 0))
locs = []
for s in range(numSteps):
f.moveDrunk(d)
locs.append(f.getLoc(d))
xVals, yVals = [], []
for loc in locs:
xVals.append(loc.getX())
yVals.append(loc.getY())
curStyle = styleChoice.nextStyle()
plt.plot(xVals, yVals, curStyle,
label = dClass.__name__)
plt.title('Spots Visited on Walk ('
+ str(numSteps) + ' steps)')
plt.xlabel('Steps East/West of Origin')
plt.ylabel('Steps North/South of Origin')
plt.legend(loc = 'best')
