python 复数归一化处理

最新推荐文章于 2024-06-05 22:03:15 发布

Alex-Leung

最新推荐文章于 2024-06-05 22:03:15 发布

阅读量1.8k

点赞数 3

分类专栏： Python 文章标签： python

本文链接：https://blog.csdn.net/a2824256/article/details/107976164

版权

Python 专栏收录该内容

58 篇文章 3 订阅

订阅专栏

python 复数归一化处理

归一化算法

复数格式为 $x = a + b i$ , 归一化算法如下:

$a\div\sqrt{a^2 + b^2} + bi\div\sqrt{a^2 + b^2}$

算法实现

from math import sqrt

def normalize(num):
	# numpy里complex可改为numpy.complex128
    if type(num) == complex:
        real = num.real
        imag = num.imag
        temp = sqrt(real*real+imag*imag)
        real = real/temp
        imag = imag/temp * 1j
        return real + imag
    else:
        return num
        
# 测试        
d = 8 + 2j
print(normalize(d)) # print (0.9701425001453319+0.24253562503633297j)