在python中操作复数

问题描述

我正在处理这个等式：

import numpy as np
ld = 1.3
aut = -np.sqrt(-4*ld*(ld - 1)*np.log(ld) + (ld + np.log(ld) - 1)**2)/(2*(ld - 1)) + (ld + np.log(ld) - 1)/(2*(ld- 1))
aut

但是当我尝试获取结果时出现错误：

invalid value encountered in sqrt

我知道解决方案可能是一个复数。有谁知道我怎样才能从这个等式中得到结果？

谢谢！

解决方法

问题是 np.sqrt 遇到负数。由于您正在寻找复杂的解决方案，一个（明显的）解决方案是将数字包裹在 np.complex 中：

(-np.sqrt(np.complex(-4*ld*(ld - 1)*np.log(ld) 
         + (ld + np.log(ld) - 1)**2 + 0))/(2*(ld - 1)) 
+ (ld + np.log(ld) - 1)/(2*(ld- 1))
)

输出：

(0.937273774112485-0.5083597988171367j)