用Python中的scimath计算复数输入的平方根

在这篇文章中，我们将介绍如何在Python中使用NumPy用scimath计算复数输入的平方根。

示例

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

NumPy.emath.sqrt 方法

NumPy库中的np.emath.sqrt()方法可以计算复数输入的平方根。与numpy.sqrt.返回NaN不同，负的输入元素会返回一个复数值。

语法: np.emath.sqrt()

参数:

x：类似对象的数组。输入数组。

返回：out:标量或ndarray。

示例 1:

如果数组包含负的输入值，输出中会返回复数，数组的形状、数据类型和尺寸可以通过.shape , .dtype , 和 .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))

输出:

[-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]

示例 2:

在这个例子中，NumPy包被导入。使用NumPy.array()方法创建了一个2维的复数，并使用np.emath.sqrt()来计算复数输入的平方根。数组的形状、数据类型和尺寸可以通过 .shape , .dtype , 和 .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))

输出:

[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]