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