Question

我写了一个类来处理三维向量，如下所示

class vector(object):

def __init__(self, x=None, y=None, z=None, angle=None):

    if angle == None:
        self.x, self.y, self.z = x, y, z
    if angle != None:
        if angle == "rad":
            self.r, self.theta, self.phi = x, y, z
        if angle == "deg":
            self.r = x
            self.theta = y * 2 * pi / 360.
            self.phi = z * 2 * pi / 360.
        self.x = self.r * sin(self.theta) * cos(self.phi)
        self.y = self.r * sin(self.theta) * sin(self.phi)
        self.z = self.r * cos(self.theta)

def write(self):
    file.write("[" + str(self.x) + ",\t" + str(self.y) + ",\t" + str(self.z) + "]")

def write_sph(self):
    file.write("[" + str(self.mag()) + ",\t" + str(self.gettheta()) + ",\t" + str(self.getphi()) + "]")

def getx(self):
    return self.x
def gety(self):
    return self.y
def getz(self):
    return self.z

def setx(self, x):
    self.x = x
def sety(self, y):
    self.y = y
def setz(self, z):
    self.z = z

def square(self):
    return self.x*self.x + self.y*self.y + self.z*self.z
def mag(self):
    return sqrt(self.square())

def gettheta(self):
    return arccos(self.z / self.mag())
def getphi(self):
    return arctan2(self.y, self.x) # sign depends on which quadrant the coordinates are in

def __add__(self, vector(other)):
    v_sum = vector(other.gettx() + self.gettx(), other.getty() + self.getty(), other.getty() + self.getty())
    return v_sum

在最后一个定义中，我试图覆盖运算符以进行添加。该定义通过调用名为other的新向量并将其x，y，z组件添加到self的相应组件来工作。当我运行代码时，我告诉定义的语法无效。如何正确定义此重写定义的向量参数？另外，将def __ add __的定义更改为简单的def add make会有什么区别？那些下划线表示什么？

Answer 1

您的参数列表中不应该有vector(other) - 只需说other即可。此外，您还需要在add方法中修复拼写错误：

def __add__(self, other):
    v_sum = vector(other.getx() + self.getx(), other.gety() + self.gety(), other.getz() + self.getz())
    return v_sum

在Python中覆盖用户定义的矢量类的plus运算符

1 个答案: