numpy.lib.scimath.sqrt

lib.scimath.sqrt(x) [source]

Compute the square root of x.

For negative input elements, a complex value is returned (unlike numpy.sqrt which returns NaN).

Parameters

xarray_like

The input value(s).

Returns

outndarray or scalar

The square root of x. If x was a scalar, so is out, otherwise an array is returned.

See also

numpy.sqrt

Examples

For real, non-negative inputs this works just like numpy.sqrt:

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

But it automatically handles negative inputs:

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