Python中的 numpy.sqrt()

numpy.sqrt(array[, out])函数用于按元素确定数组的正平方根。

Syntax: numpy.sqrt()

Parameters:
array : [array_like] Input values whose square-roots have to be determined.
out : [ndarray, optional] Alternate array object in which to put the result; if provided, it must have the same shape as arr.

Returns : [ndarray] Returns the square root of the number in an array.

代码#1：

# Python program explaining
# numpy.sqrt() method 
  
# importing numpy
import numpy as geek 
  
# applying sqrt() method on integer numbers 
arr1 = geek.sqrt([1, 4, 9, 16])
arr2 = geek.sqrt([6, 10, 18])
  
print("square-root of an array1  : ", arr1)
print("square-root of an array2  : ", arr2)

输出：

square-root of an array1  :  [ 1.  2.  3.  4.]
square-root of an array2  :  [ 2.44948974  3.16227766  4.24264069]

代码#2：

# Python program explaining
# numpy.sqrt() method 
  
# importing numpy
import numpy as geek 
  
# applying sqrt() method on complex numbers
arr = geek.sqrt([4, -1, -5 + 9J])
  
print("square-root of an array  : ", arr)

输出：

square-root of an array  :  [ 2.00000000+0.j  0.00000000+1.j  1.62721083+2.76546833j]

代码#3：

# Python program explaining
# numpy.sqrt() method 
  
# importing numpy
import numpy as geek 
  
# applying sqrt() method on negative element of real numbers 
arr = geek.sqrt([-4, 5, -6])
  
print("square-root of an array  : ", arr)

输出：

RuntimeWarning: invalid value encountered in sqrt
square-root of an array  :  [ nan  2.23606798  nan]