Calcule la raíz cuadrada de entradas complejas con scimath en Python

En este artículo, cubriremos cómo calcular la raíz cuadrada de entradas complejas con scimath en Python usando NumPy .

Ejemplo

Input: [-1 -2]
Output: [0.+1.j  0.+1.41421356j]
Explanation: Square root of complex input.

Método NumPy.emath.sqrt

El método np.emath.sqrt() de la biblioteca NumPy calcula la raíz cuadrada de entradas complejas. Se devuelve un valor complejo para elementos de entrada negativos a diferencia de numpy.sqrt. que devuelve NaN.

Sintaxis: np.emath.sqrt()

Parámetros:

• x: array como objeto. array de entrada.

Retorno: salida: escalar o ndarray.

Ejemplo 1:

Si la array contiene valores de entrada negativos, se devuelven números complejos en la salida y la forma, el tipo de datos y las dimensiones de la array se pueden encontrar mediante . forma , .dtype y . atributos ndim .

import numpy as np
  
# Creating an array
array = np.array([-1, -2])
print(array)
  
# shape of the array is
print("Shape of the array is : ",array.shape)
  
# dimension of the array
print("The dimension of the array is : ",array.ndim)
  
# Datatype of the array
print("Datatype of our Array is : ",array.dtype)
  
# computing the square root of negative inputs
print(np.emath.sqrt(array))

Producción:

[-1 -2]
Shape of the array is :  (2,)
The dimension of the array is :  1
Datatype of our Array is :  int64
[0.+1.j  0.+1.41421356j]

Ejemplo 2:

En este ejemplo, se importa el paquete NumPy. Se crea una array compleja de 2 dimensiones con el método NumPy.array() y se usa np.emath.sqrt() para calcular la raíz cuadrada de las entradas complejas. La forma, el tipo de datos y las dimensiones de la array se pueden encontrar mediante . forma , .dtype y . atributos ndim .

import numpy as np
  
# Creating an array
array = np.array([complex(1,2),complex(3,5)])
print(array)
  
# shape of the array is
print("Shape of the array is : ",array.shape)
  
# dimension of the array
print("The dimension of the array is : ",array.ndim)
  
# Datatype of the array
print("Datatype of our Array is : ",array.dtype)
  
# computing the square root of complex inputs
print(np.emath.sqrt(array))

Producción:

[1.+2.j 3.+5.j]
Shape of the array is :  (2,)
The dimension of the array is :  1
Datatype of our Array is :  complex128
[1.27201965+0.78615138j 2.10130339+1.18973776j]