如何从具有复数的数组中获得具有模数的数组？

Question

我在复杂数字的数组可视化方面遇到了问题。我需要一个具有复数模数的数组来制作一个情节，但我不知道如何使用numpy来做。我该怎么办？

Answer 1

如果m是一个复数数组，您可以使用numpy.abs(m)

获取每个元素的模数

举个例子：

>>> m=np.random.random((2,2))+np.random.random((2,2))*1j
>>> m
array([[ 0.95274284+0.39078475j,  0.80123497+0.10182257j],
   [ 0.50777149+0.41198386j,  0.83248958+0.20489083j]])
>>> numpy.abs(m)
array([[ 1.02977262,  0.80767896],
   [ 0.65388271,  0.85733258]])