数字j是复数:
>>> 1j * 1j
(-1+0j)
Run Code Online (Sandbox Code Playgroud)
您可以使用real,imag属性提取实部和虚部:
>>> c = 1 + 2j
>>> c.real
1.0
>>> c.imag
2.0
Run Code Online (Sandbox Code Playgroud)
将带有复数的 numpy 数组转换为带有浮点数的数组:相同;使用real,imag属性。
>>> np.array([1+0j, 2+0j]).real
array([ 1., 2.])
Run Code Online (Sandbox Code Playgroud)