使用 __add__ 方法相加

Question

我目前正在研究一个创建球并为它们提供位置的课程。我想为类创建一个add方法，该方法获取两个球的位置并将它们加在一起。我正在使用的代码是：

class Ball:

    def __init__ (self, x, y):

        self.position = [x, y]

    def __add__ (self, other):

        return self.position + other.position

    

ball1 = Ball(3, 4)

print (ball1.position)

ball2 = Ball(5, 8)

print (ball2.position)

ball3 = Ball(4, 4)

print (ball3.position)

ball4 = ball1 + ball3

print (ball4)

代码现在的工作方式并不符合预期。我希望 ball1 + ball3 位置相加，但我得到的打印结果如下：

[3, 4]

[5, 8]

[4, 4]

[3, 4, 4, 4]

我们将 ball1 和 ball3 的 x 和 y 值并排放置，而不是相加。

Answer 1

当您将两个列表添加在一起时，它只是简单地附加。您的添加需要如下所示：def __add__ (self, other):     return [self.position[0]+other.position[0], self.position[1]+other.position[1]]

Answer 2

使用数组时，“+”运算符连接两个操作数中的元素并返回一个新数组。所以plus并没有按照你想象的那样做。您必须开发另一个函数来对数组的每个成员求和并返回这个新数组

Answer 3

这是因为您在 2 个 python 列表之间使用求和运算符，这将连接两个列表。如果你想按元素求和，你必须这样做：return [self.x+other.x, self.y+other.y]我还建议您查看numpy库，它将提供您想要的数学运算符。在这种情况下，您的类可以重写为：import numpy as npclass Ball:    def __init__ (self, x, y):        self.position = np.array([x, y])    def __add__ (self, other):        return self.position + other.position    ball1 = Ball(3, 4)print (ball1.position)ball2 = Ball(5, 8)print (ball2.position)ball3 = Ball(4, 4)print (ball3.position)ball4 = ball1 + ball3print (ball4)结果：>>> ball1 = Ball(3, 4)>>> print (ball1.position)[3 4]>>> ball2 = Ball(5, 8)>>> print (ball2.position)[5 8]>>> ball3 = Ball(4, 4)>>> print (ball3.position)[4 4]>>> ball4 = ball1 + ball3>>> print (ball4)[7 8]

Answer 4

通过列表理解单独添加项目并zip使用：[b1 + b3 for b1, b3 in zip(ball1.position, ball3.position)]class Ball:    def __init__ (self, x, y):        self.position = [x, y]    def __add__ (self, other):        return self.position + other.positionball1 = Ball(3, 4)print (ball1.position)ball2 = Ball(5, 8)print (ball2.position)ball3 = Ball(4, 4)print (ball3.position)ball4 = [b1 + b3 for b1, b3 in zip(ball1.position, ball3.position)]print (ball4)[3, 4][5, 8][4, 4][7, 8]编辑：您可以进一步简化列表理解：ball4 = [sum(b) for b in zip(ball1.position,ball3.position)]