Как лучше всего преобразовать сложный массив numpy в массив величин?

Question

Я имею в виду что-то вроде:

mag(complex_array) = [ sqrt(complex.real * complex.real + complex.imag + complex.imag) for complex in complex_array ]

Есть ли для этого встроенная функция?

Answer 1

Просто используйте np.abs:

>>> a = np.array([1+2j, 3+4j])
>>> np.abs(a)
array([2.23606798, 5.        ])