编写一个复数类,包含实部和虚部,能够对复数进行加减运算,在main函数中进行调用
时间: 2024-10-24 09:04:18 浏览: 20
在Python中,我们可以创建一个名为`ComplexNumber`的类,用于表示复数并支持基本的加法和减法操作。下面是一个简单的实现:
```python
class ComplexNumber:
def __init__(self, real, imag):
self.real = real
self.imaginary = imag
# 加法运算
def add(self, other):
return ComplexNumber(self.real + other.real, self.imaginary + other.imaginary)
# 减法运算
def subtract(self, other):
return ComplexNumber(self.real - other.real, self.imaginary - other.imaginary)
# 主函数
def main():
# 创建两个复数对象
num1 = ComplexNumber(3, 4)
num2 = ComplexNumber(-2, 5)
# 进行加法和减法操作
sum_result = num1.add(num2)
diff_result = num1.subtract(num2)
# 打印结果
print(f"Sum: {sum_result.real} + {sum_result.imaginary}i")
print(f"Difference: {diff_result.real} + {diff_result.imaginary}i")
if __name__ == "__main__":
main()
```
在这个例子中,我们首先定义了一个`ComplexNumber`类,它有两个属性(实部和虚部),以及对应的加法(`add`)和减法(`subtract`)方法。在`main`函数中,我们创建了两个复数对象,并通过调用这两个方法演示了它们的使用。
阅读全文