Python RandomWalk не ходить

Question

Мой профессор дал мне этот код, и единственная часть, которую я собираюсь сделать, это область TODO. Я все еще новичок в Python и никогда не затрагивал этот тип проекта, поэтому я в замешательстве. все, для чего предназначен этот проект - получить изображение графического графика.

import math, random, pylab
class Location(object):
    def __init__(self, x, y):
        self.x = float(x)
        self.y = float(y)
    def move(self, xc, yc):
        return Location(self.x+float(xc), self.y+float(yc))
    def getCoords(self):
        return self.x, self.y
    def getDist(self, other):
        ox, oy = other.getCoords()
        xDist = self.x - ox
        yDist = self.y - oy
        return math.sqrt(xDist**2 + yDist**2)
class CompassPt(object):
    possibles = ('N', 'S', 'E', 'W')
    def __init__(self, pt):
        if pt in self.possibles: self.pt = pt
        else: raise ValueError('in CompassPt.__init__')
    def move(self, dist):
        if self.pt == 'N': return (0, dist)
        elif self.pt == 'S': return (0, -dist)
        elif self.pt == 'E': return (dist, 0)
        elif self.pt == 'W': return (-dist, 0)
        else: raise ValueError('in CompassPt.move')
class Field(object):
    def __init__(self, drunk, loc):
        self.drunk = drunk
        self.loc = loc
    def move(self, cp, dist):
        oldLoc = self.loc
        xc, yc = cp.move(dist)
        self.loc = oldLoc.move(xc, yc)
    def getLoc(self):
        return self.loc
    def getDrunk(self):
        return self.drunk
class Drunk(object):
    def __init__(self, name):
        self.name = name
    def move(self, field, time = 1):
        if field.getDrunk() != self:
            raise ValueError('Drunk.move called with drunk not in field')
        for i in range(time):
            pt = CompassPt(random.choice(CompassPt.possibles))
            field.move(pt, 1)

вот часть, которую я должен сделать и пытался сделать. пока мой график не движется лол. Я получаю точку в середине графика, если только это не происходит случайным образом в одном месте, если это возможно, лол.

# TODO TODO TODO TODO TODO TODO TODO TODO TODO TODO TODO TODO
def performTrial(time, f):
    start = f.getLoc()
    # TODO
    distances = [0.0] 
    for t in range(1, time + 1):
        # TODO
        f.getDrunk().move(f) 
        newLoc = f.getLoc() 
        distance = newLoc.getDist(start) 
        distances.append(distance) 
    xcoords, ycoords = distances[::2], distances[1::2]
    return xcoords,ycoords
# END OF TODOEND OF TODO END OF TODO END OF TODO END OF TODO

остальное здесь также предоставлено профессором

drunk = Drunk('Homer Simpson')
for i in range(1):
    f = Field(drunk, Location(0, 0))
    coords = performTrial(500, f)
    print(coords)
    pylab.plot(coords[0],coords[1],marker='^',linestyle=':',color='b')
pylab.title('Homer\'s Random Walk')
pylab.xlabel('Time')
pylab.ylabel('Distance from Origin')
pylab.show()

обновление: исправлены координаты, но теперь я получаю сообщение об ошибке:

in _xy_from_xy(self, x, y)
    235         y = np.atleast_1d(y)
    236         if x.shape[0] != y.shape[0]:
--> 237             raise ValueError("x and y must have same first dimension")
    238         if x.ndim > 2 or y.ndim > 2:
    239             raise ValueError("x and y can be no greater than 2-D")

ValueError: x and y must have same first dimension 

Answer 1

def move(self, xc, yc):
    return Location(self.x+float(xc), self.y+float(yc))

это ничего не перемещает, оно просто возвращает новые координаты. Также не нужно приводить xc и yc к float.

Если эта функция предназначена для перемещения местоположения, вы должны добавить это:

def move(self, xc, yc):
    self.x += xc
    self.y += yc
    return Location(self.x, self.y)