欢迎访问宙启技术站
智能推送

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_addcomplex_sub两个函数来实现复数的加法和减法。这两个函数分别计算实部和虚部的和或差,并使用complex函数构造出新的复数对象。

然后,通过创建两个复数对象ab,分别表示第一个和第二个复数。接下来,将这两个复数对象作为参数传递给complex_addcomplex_sub函数进行计算,得到结果cd

最后,通过使用print函数输出结果。