从函数返回一个复数

Question

我想为变量分配一个虚数:

import math

a = sqrt(4)j
print(a)

这会导致SyntaxError:

Line 3: SyntaxError: bad input ('j')

我可以像这样给变量分配一个虚数:

a = 2j

我该如何解决这个问题？

Answer 1

你可以简单地complex()用来返回:

real + imag*1j 或将字符串或数字转换为复数

>>> a = complex(0,math.sqrt(4))
>>> a
2j