一、复数的数学运算
复数可以用使用函数 complex(real, imag) 或者是带有后缀j的浮点数来指定
a=complex(2,4)
print(a) # (2+4j)
b=2-5j
# 获取对应的实部、虚部和共轭复数
print(b.real,b.imag,b.conjugate()) # 2.0 -5.0 (2+5j)
# 数学运算
print(a+b) # (4-1j)
print(abs(a)) # 4.47213595499958
复数函数比如正弦、余弦或平方根,使用 cmath 模块
import cmath
print(cmath.sin(a))
print(cmath.cos(a))
print(cmath.exp(a))
# (24.83130584894638-11.356612711218173j)
# (-11.36423470640106-24.814651485634183j)
# (-4.829809383269385-5.5920560936409816j)
使用 numpy 也可以处理复数
# 构造复数数组
import numpy as np
c=np.array([1+2j,2+3j,3+4j])
print(c) # [ 1.+2.j 2.+3.j 3.+4.j]
print(c+2) # [ 3.+2.j 4.+3.j 5.+4.j]
print(np.sin(c)) # [ 3.16577851 +1.95960104j 9.15449915 -4.