Как получить массив с модулем из массива с комплексными числами?

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.8.10182257j], 
    [ 0.50777149+0.41198386j, 0.83248958+0.20489083j]]) 
>>> numpy.abs(m) 
array([[ 1.02977262, 0.80767896], 
    [ 0.65388271, 0.85733258]])