Функция imag
Функция imag возвращает мнимую часть комплексного числа.
Если передано вещественное число, результатом будет 0.
Если передан массив, функция применяется поэлементно и возвращает массив мнимых частей.
Функция полезна при работе с комплексными данными в научных расчетах и обработке сигналов.
Синтаксис
np.imag(arr)
Пример
Давайте извлечем мнимую часть из комплексного числа:
import numpy as np
res = np.imag(3 + 4j)
print(res)
Результат выполнения кода:
4.0
Пример
Применим функцию imag к массиву комплексных чисел:
import numpy as np
arr = np.array([1 + 2j, 3 - 4j, 5 + 6j, 7 - 8j])
res = np.imag(arr)
print(res)
Результат выполнения кода:
[ 2. -4. 6. -8.]
Пример
Если передан вещественный массив, результатом будет массив из нулей:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
res = np.imag(arr)
print(res)
Результат выполнения кода:
[0. 0. 0. 0. 0.]
Пример
Функцию imag можно применить к многомерному массиву комплексных чисел:
import numpy as np
arr = np.array([[1 + 1j, 2 + 2j], [3 + 3j, 4 + 4j]])
res = np.imag(arr)
print(res)
Результат выполнения кода:
[[1. 2.]
[3. 4.]]