Python numpy scimath.sqrt用法及代码示例

本文简要介绍 python 语言中 numpy.lib.scimath.sqrt 的用法。

用法:

lib.scimath.sqrt(x)

计算 x 的平方根。

对于负输入元素，返回一个复数值(与返回 NaN 的 numpy.sqrt 不同)。

参数：

x： array_like

输入值。

返回：

out： ndarray 或标量

x 的平方根。如果 x 是标量，则输出，否则返回数组。

例子：

对于真实的非负输入，这就像 numpy.sqrt 一样工作：

>>> np.emath.sqrt(1)
1.0
>>> np.emath.sqrt([1, 4])
array([1.,  2.])

但它会自动处理负输入：

>>> np.emath.sqrt(-1)
1j
>>> np.emath.sqrt([-1,4])
array([0.+1.j, 2.+0.j])