自定义一个复数类,实现四则运算。
代码语言:javascript复制import math
class Complex():
def __init__(self, real, imaginary):
self.real = real
self.imaginary = imaginary
def __add__(self, no):
real = self.real no.real
imaginary = self.imaginary no.imaginary
return Complex(real, imaginary)
def __sub__(self, no):
real = self.real - no.real
imaginary = self.imaginary - no.imaginary
return Complex(real, imaginary)
def __mul__(self, no):
real = self.real * no.real - self.imaginary * no.imaginary
imaginary = self.real * no.imaginary self.imaginary * no.real
return Complex(real, imaginary)
def __truediv__(self, no):
x = float(no.real ** 2 no.imaginary ** 2)
y = self * Complex(no.real, -no.imaginary)
real = y.real / x
imaginary = y.imaginary / x
return Complex(real, imaginary)
def mod(self):
real = math.sqrt(self.real ** 2 self.imaginary ** 2)
return Complex(real, 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')
有个巨坑的地方:python3定义一般的除法要用__truediv__
而不是__div__