python建立复数数组_Python之复数、分数、大型数组数学运算（complex、cmath、numpy、fraction...

一、复数的数学运算

复数可以用使用函数 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.