矢量添加不工作

时间:2017-09-28 03:00:28

标签: python

我写了这段代码来添加矢量。乍一看,它看起来工作正常,但我发现在某些附加问题中,矢量添加顺序实际上很重要,这不应该发生。例如,给定v1 = Vector(4, 127) v2 = Vector(3, 37) v3 = Vector(2, 290) v4 = Vector(6, 190)print(v1 + v2 + v3 + v4)会产生与print(v1 + v4 + v2 + v3)不同的结果 有人可以指出我的错误并解释为什么它错了吗?

import math

class Vector():
    def __init__(self, magnitude, direction):
        self.magnitude = round(magnitude, 4)
        self.direction = round(direction % 360, 4)
        self.xc = self.GetXComponent()
        self.yc = self.GetYComponent()

    def __str__(self):
        return "{} at {}°".format(self.magnitude, self.direction)

    def __add__(self, other):
        if other.__class__ == Vector:
            xc = self.xc + other.xc
            yc = self.yc + other.yc
            magnitude = round(math.sqrt(xc ** 2 + yc ** 2), 4)
            direction = round(math.degrees(math.atan(yc / xc)), 4)
            return Vector(magnitude, direction)
        else:
            raise TypeError

    def GetXComponent(self):
        return round(self.magnitude * math.cos(math.radians(self.direction)), 4)

    def GetYComponent(self):
        return round(self.magnitude * math.sin(math.radians(self.direction)), 4)

1 个答案:

答案 0 :(得分:3)

math.atan(x)的数学范围(输出)范围仅为π(180°,半圈)。如果您添加的任何结果矢量位于象限II或III(x为负数),则它们将围绕圆圈旋转一半。

使用math.atan2(y, x)表示完整的可能范围。如果您的90/270°矢量现在会导致ZeroDivisionError,那么这也将消除您将获得的错误。

其他评论:

  • 当您想要输出值时,回合一次。这是a)更多的代码来围绕无处不在,以及b)不太精确。

  • 在内部使用自然单位,即弧度或笛卡尔坐标,可能更容易,同时仍然接受/输出您想要的任何内容。