数组中复数的大小

Magnitude of Complex Numbers in Array

我有以下数组:

complex = [4+1j, 4+ 0j , 4 + 2j]

有没有一种有效的方法来转换为幅度(像这个伪代码):

mag = np.magnitude(complex) = [sqrt(17), 4, sqrt(20)]

谢谢

你想要numpy.abs:

np.abs([4+1j, 4+ 0j , 4 + 2j])

输出:

array([4.12310563, 4.        , 4.47213595])
# == array([sqrt(17), sqrt(16), sqrt(20)])