Python中实现复数加法和减法的示例代码
发布时间:2024-01-06 19:44:29
复数在Python中可以使用complex类进行表示。complex类有两个属性,real表示实部,imag表示虚部。
要实现复数的加法和减法,只需使用+和-运算符即可。
下面是一个示例代码:
# 复数加法
def complex_add(a, b):
real_part = a.real + b.real
imag_part = a.imag + b.imag
return complex(real_part, imag_part)
# 复数减法
def complex_sub(a, b):
real_part = a.real - b.real
imag_part = a.imag - b.imag
return complex(real_part, imag_part)
# 使用示例
a = complex(1, 2) # 第一个复数,实部为1,虚部为2
b = complex(3, 4) # 第二个复数,实部为3,虚部为4
# 复数加法
c = complex_add(a, b)
print(f"复数加法结果为:{c}")
# 复数减法
d = complex_sub(a, b)
print(f"复数减法结果为:{d}")
输出结果为:
复数加法结果为:(4+6j) 复数减法结果为:(-2-2j)
在示例代码中,通过定义complex_add和complex_sub两个函数来实现复数的加法和减法。这两个函数分别计算实部和虚部的和或差,并使用complex函数构造出新的复数对象。
然后,通过创建两个复数对象a和b,分别表示第一个和第二个复数。接下来,将这两个复数对象作为参数传递给complex_add和complex_sub函数进行计算,得到结果c和d。
最后,通过使用print函数输出结果。
