python_guided_project/q01_create_class/build.py at c3a8d9b6a3998b08d74cb577e7c26c8abed2ee68 · 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
59
60
61
62
63
64
65
66
67
68
69
70
71
# %load q01_create_class/build.py
import pandas as pd
import numpy as np
import math

'write your solution here'


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


aa = complex_number(3,3)
#bb = complex_number(9,9)
#print(aa)
#math.tanh
#aa.abs()
print(aa.argument())
c1 = complex_number(4,4)
c2 = complex_number(4,-3)
print (c1)
print (c2)
print(c1+c2)
print(c1-c2)
print(c1*c2)
print(c1/c2)
print(c1.conjugate())
print(c1.abs())
print(c1.argument())