尝试重载运算符“/”时出错

时间:2014-02-18 16:19:21

标签: python python-3.x operator-overloading vector-graphics

我最近开始自学游戏编程。有人建议我从Python开始,我得到了“用Python和Pygame开始游戏开发:从新手到专业”的书。我得到了一个他们教导Vector和创建Vector2类的部分。一切顺利,直到我试图超过除法运算符。 我的代码是这样的:

class Vector2(object):

  def __init__(self, x=0.0, y=0.0):
    self.x = x
    self.y = y

  def __str__(self):
    return "(%s, %s)"%(self.x, self.y)

  @classmethod
  def from_points(cls, P1, P2):
    return cls(P2[0] - P1[0], P2[1] - P1[1])

  def __add__(self,rhs):
    return Vector2(self.x + rhs.x, self.y + rhs.y)

  def __sub__(self,rhs):
    return Vector2(self.x - rhs.x, self.y - rhs.y)

  def __mul__(self, scalar):
    return Vector2( self.x*scalar, self.y*scalar)

  def __div__(self, scalar):
    return Vector2( self.x/scalar, self.y/scalar)

现在,当我尝试调用“/”运算符时,会显示:

AB = Vector2(10.0,25.0)
print(AB)   # <<<<(10.0, 25.0)
v1 = AB + Vector2(20.,10.)
print(v1)   # <<<<(30.0, 35.0)
v2 = AB - Vector2(20.,10.)
print(v2)   # <<<<(-10.0, 15.0)
v3 = AB * 3
print(v3)   # <<<<(30.0, 75.0)
print(v3 / 3)
TypeError: unsupported operand type(s) for /: 'Vector2' and 'int'

这完全在Python 3.3中,但如果我使用Python 2.7运行它,一切正常。 问题出在哪里?

2 个答案:

答案 0 :(得分:46)

在Python 3.x中,您需要重载__floordiv____truediv__运算符,而不是__div__运算符。前者对应于//操作(返回一个整数),后者对应/(返回一个浮点数)。

答案 1 :(得分:5)

在Python 3中,除法运算符称为__truediv____floordiv__。有关详细信息,请参阅Data model文档。