python_guided_project/q01_create_class/build.py at 73ccc6f0423c5e7bbc078f0d0846fc7d6c5f68e4 · commit-live-students/python_guided_project · GitHub

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
# %load q01_create_class/build.py


import pandas as pd
import numpy as np
import math

# https://www.python-course.eu/python3_magic_methods.php

class complex_number:
    def __init__(self, real = 0, imag = 0):
        self.real = real
        self.imag = imag

    def __str__(self):
        if self.imag<0:
            return '{}-i{}'.format(self.real,abs(self.imag))
        else:
            return '{}+i{}'.format(self.real,self.imag)

    def __add__(self,other):
        c = self.real+other.real
        d = self.imag+other.imag
        return complex_number(c,d)

    def __sub__(self,other):
        c= self.real-other.real
        d= self.imag-other.imag
        return complex_number(c,d)

    def __mul__(self,other):
        c= (self.real*other.real - self.imag*other.imag)
        d= (self.real*other.imag + self.imag*other.real)
        return complex_number(c,d)

    def __truediv__(self,other):
        c= ((self.real*other.real+self.imag*other.imag))/(other.real**2+other.imag**2)
        d= ((self.imag*other.real-self.real*other.imag))/(other.real**2+other.imag**2)
        #return complex_number(c,d)
        return c,d

    def conjugate(self):
        c = self.real
        d = self.imag
        return complex_number(c,-d)

    def abs(self):
        c = (self.real**2+self.imag**2)**(1/2)
        return c

    def argument(self):
        c = math.degrees(np.arctan(self.real/self.imag))
        return c