FilesExpand file tree

 master

/

vector2d.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

55 lines (37 loc) · 1010 Bytes

 master

/

vector2d.py

Copy path

Top

File metadata and controls

• Code
• Blame

55 lines (37 loc) · 1010 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

"""

vector2d.py: a simplistic class demonstrating some special methods

It is simplistic for didactic reasons. It lacks proper error handling,

especially in the ``__add__`` and ``__mul__`` methods.

This example is greatly expanded later in the book.

Addition::

>>> v1 = Vector(2, 4)

>>> v2 = Vector(2, 1)

>>> v1 + v2

Vector(4, 5)

Absolute value::

>>> v = Vector(3, 4)

>>> abs(v)

5.0

Scalar multiplication::

>>> v * 3

Vector(9, 12)

>>> abs(v * 3)

15.0

"""

import math

class Vector:

def __init__(self, x=0, y=0):

self.x = x

self.y = y

def __repr__(self):

return f'Vector({self.x!r}, {self.y!r})'

def __abs__(self):

return math.hypot(self.x, self.y)

def __bool__(self):

return bool(abs(self))

def __add__(self, other):

x = self.x + other.x

y = self.y + other.y

return Vector(x, y)

def __mul__(self, scalar):

return Vector(self.x * scalar, self.y * scalar)