-
Notifications
You must be signed in to change notification settings - Fork 0
/
complex_number.py
117 lines (68 loc) · 2.97 KB
/
complex_number.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
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
104
105
106
107
108
109
110
111
112
113
import numpy as np
import matplotlib.pyplot as plt
import math
class ComplexNumber:
#A complex number representation
def __init__(self, real, imaginary):
self.real = real
self.imaginary = imaginary
self.mod = math.sqrt(self.real*self.real + self.imaginary*self.imaginary)
self.theta = math.atan(self.imaginary/self.real)
def copy(self):
return ComplexNumber(self.real, self.imaginary)
def __str__(self):
return "{} + {}i".format(self.real, self.imaginary)
def getImaginary(self):
return self.imaginary
def getReal(self):
return self.real
def conjugate(self):
return ComplexNumber(self.real, -self.imaginary)
def __add__(self, other):
# Making sure that the other is a complex number
assert isinstance(other, ComplexNumber)
return ComplexNumber((self.real+other.real),(self.imaginary+other.imaginary))
def __sub__(self, other):
# Making sure that the other is a complex number
assert isinstance(other, ComplexNumber)
return self + (ComplexNumber(-other.real, -other.imaginary))
def __mul__(self, other):
# Making sure that the other is a complex number
assert isinstance(other, ComplexNumber)
#Use public varadic parameters to accept public vary of input data types
return ComplexNumber((self.real*other.real - self.imaginary*other.imaginary), (self.real*other.imaginary + other.real*self.imaginary))
# Improve runtime
def __pow__(self, other):
# Making sure that exponent is integer
assert(other, int)
s = self.copy()
t = self.copy()
i = 1
while i < other :
s = t*s
i+=1
return s
def __truediv__(self, other):
# Making sure that the other is a complex number
assert isinstance(other, ComplexNumber)
denom = other.real*other.real - other.imaginary*other.imaginary
num = self * other.conjugate()
return ComplexNumber(num.real/denom, num.imaginary/denom)
def polar_form(self):
return "{} ( cos{} + isin{})".format(self.mod, self.theta, self.theta)
def eForm(self):
return "{}e^(i{}))".format(self.mod, self.theta)
def infSeq(self):
t = self.copy()
s = ComplexNumber(1,0)
for i in range(100):
t = ComplexNumber(t.real * t.real - t.imaginary * t.imaginary,2*t.real*t.imaginary) + s
print(t)
if (t.real * t.real + t.imaginary * t.imaginary) >= 4 :
return i
return 0
def plot(self):
plt.plot(self.real, self.imaginary, linestyle='--', marker='o', color='b')
plt.xlabel("Real")
plt.ylabel("Imaginary")
plt.show()