数组中复数的大小

如何解决数组中复数的大小

我有以下数组：

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)])