python_guided_project/q01_create_class/build.py at 6840d0f5d9d49cd19541e63fdaa3130f4bbe545f · 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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# %load q01_create_class/build.py
import pandas as pd
import numpy as np
import math

'write your solution here'


class complex_number:
    '''The complex number class.

    Attributes:
    attr1 (x): Real part of complex number.
    attr2 (y): Imaginary part of complex number.

    '''
    def __init__(self, real, imag):
        self.real = real
        self.imag = imag

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

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

    def __mul__(self, other):

        if(type(self) is type(other)):
            real_part = (self.real * other.real) - (self.imag * other.imag)
            imag_part = (self.imag * other.real) + (self.real * other.imag)
        else:
            real_part = self.real * other
            imag_part = self.imag * other

        return complex_number(real_part, imag_part)

    def __str__(self):
        #print('Test: ', self.imag)
        if(self.imag > 0):
            if(self.real == 0):
                return 'i{0}'.format(self.imag)
            else:
                return '{0} + i{1}'.format(self.real, self.imag)

        elif (self.imag == 0):
            if(self.real == 0):
                return ' '
            else:
                return '{0}'.format(self.real)

        else:
            if(self.real == 0):
                return '-i{0}'.format(abs(self.imag))
            else:
                return '{0} - i{1}'.format(self.real, abs(self.imag))

    def conjugate(self):
        if(self.imag > 0):
            return  complex_number(self.real, (self.imag * -1))
        else:
            return complex_number(self.real, abs(self.imag))

    def __truedivWithoutConjug__(self, other):
        numerator_real = (self.real * other.real) + (self.imag * other.imag)
        numerator_imag = (self.imag * other.real) - (self.real * other.imag)
        denominator = (other.real**2) + (other.imag**2)
        return (numerator_real/denominator, numerator_imag/denominator)

    def __truediv__(self, other):
        numerator = self * other.conjugate()
        denominator = (other.real ** 2) + (other.imag **2)
        #print('Denom is: ', denominator)
        return (numerator.real/denominator, numerator.imag/denominator)
        #return complex_number(numerator_real/denominator, numerator_imag/denominator)

    def abs(self):
        return math.sqrt((self.real ** 2) + (self.imag ** 2))

    def argument(self):
        tan_res = math.atan2(self.imag,self.real)
        return (tan_res * (180/math.pi))

a = complex_number(4,4)
print('First complex number is: ', str(a))
b = complex_number(4,-3)
print('Second complex number is: ', str(b))
add_res = a + b
print('Addition of complex numbers is: ', str(add_res))
sub_res = a - b
print('Subtraction of complex numbers is: ', str(sub_res))
mult_res = a * b
print('Multiplication of complex numbers is: ', str(mult_res))
conjug_res = b.conjugate()
print('Conjugation of complex number is: ', str(conjug_res))
div_res = a.__truediv__(b)
print('Division of complex numbers is: ', str(div_res))
abs_res = a.abs()
print('Absolute of complex number is: ', str(abs_res))
arg_res = a.argument()
print('Argument of complex number is: ', str(arg_res))
div_result = a.__truedivWithoutConjug__(b)
print('Division of complex numbers is: ', str(div_result))