You are viewing a single comment's thread. Return to all comments →
import math class Complex(object): def __init__(self, real, imaginary): self.real = real self.imaginary = imaginary def __add__(self, no): return Complex(self.real + no.real, self.imaginary + no.imaginary) def __sub__(self, no): return Complex(self.real - no.real, self.imaginary - no.imaginary) def __mul__(self, no): realpart = self.real*no.real - self.imaginary*no.imaginary imaginarypart = self.real*no.imaginary + no.real*self.imaginary return Complex(realpart, imaginarypart) def __truediv__(self, no): denom = no.real**2 + no.imaginary**2 realpart = (self.real*no.real + self.imaginary*no.imaginary)/denom imaginarypart = (no.real*self.imaginary-self.real*no.imaginary)/denom return Complex(realpart, imaginarypart) def mod(self): return Complex(math.sqrt(self.real**2 + self.imaginary**2), 0) def __str__(self): if self.imaginary == 0: result = "%.2f+0.00i" % (self.real) elif self.real == 0: if self.imaginary >= 0: result = "0.00+%.2fi" % (self.imaginary) else: result = "0.00-%.2fi" % (abs(self.imaginary)) elif self.imaginary > 0: result = "%.2f+%.2fi" % (self.real, self.imaginary) else: result = "%.2f-%.2fi" % (self.real, abs(self.imaginary)) return result if __name__ == '__main__': c = map(float, input().split()) d = map(float, input().split()) x = Complex(*c) y = Complex(*d) print(*map(str, [x+y, x-y, x*y, x/y, x.mod(), y.mod()]), sep='\n')
Seems like cookies are disabled on this browser, please enable them to open this website
Classes: Dealing with Complex Numbers
You are viewing a single comment's thread. Return to all comments →