我写了这段代码来添加矢量。乍一看,它看起来工作正常,但我发现在某些附加问题中,矢量添加顺序实际上很重要,这不应该发生。例如,给定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)
答案 0 :(得分:3)
math.atan(x)
的数学范围(输出)范围仅为π(180°,半圈)。如果您添加的任何结果矢量位于象限II或III(x为负数),则它们将围绕圆圈旋转一半。
使用math.atan2(y, x)
表示完整的可能范围。如果您的90/270°矢量现在会导致ZeroDivisionError
,那么这也将消除您将获得的错误。
其他评论:
当您想要输出值时,回合一次。这是a)更多的代码来围绕无处不在,以及b)不太精确。
在内部使用自然单位,即弧度或笛卡尔坐标,可能更容易,同时仍然接受/输出您想要的任何内容。