-
Notifications
You must be signed in to change notification settings - Fork 0
/
vectors.py
76 lines (65 loc) · 1.93 KB
/
vectors.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
# vectors.py
import math
class Vector:
def __init__(self, x=0.0, y=0.0, z=0.0):
self.x = x
self.y = y
self.z = z
def __repr__(self):
return f"Vector({self.x}, {self.y}, {self.z})"
def __str__(self):
return f"{self.x}i + {self.y}j + {self.z}k"
def __getitem__(self, item):
if item == 0:
return self.x
elif item == 1:
return self.y
elif item == 2:
return self.z
else:
raise IndexError("There are only three elements in the vector")
def __add__(self, other):
return Vector(
self.x + other.x,
self.y + other.y,
self.z + other.z,
)
def __sub__(self, other):
return Vector(
self.x - other.x,
self.y - other.y,
self.z - other.z,
)
def __mul__(self, other):
if isinstance(other, Vector): # Vector dot product
return (
self.x * other.x
+ self.y * other.y
+ self.z * other.z
)
elif isinstance(other, (int, float)): # Scalar multiplication
return Vector(
self.x * other,
self.y * other,
self.z * other,
)
else:
raise TypeError("operand must be Vector, int, or float")
def __truediv__(self, other):
if isinstance(other, (int, float)):
return Vector(
self.x / other,
self.y / other,
self.z / other,
)
else:
raise TypeError("operand must be int or float")
def get_magnitude(self):
return math.sqrt(self.x ** 2 + self.y ** 2 + self.z ** 2)
def normalize(self):
magnitude = self.get_magnitude()
return Vector(
self.x / magnitude,
self.y / magnitude,
self.z / magnitude,
)