问题描述
我有以下数组：
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)])
