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