Python Numpy.sqrt() - 平方根

  1. numpy.sqrt() 语法
  2. 示例代码:numpy.sqrt()
  3. 示例代码:numpy.sqrt(),参数为 out
  4. 示例代码:numpy.sqrt() 与负数的关系
  5. 示例代码:numpy.sqrt() 与复数一起使用

Numpy.sqrt() 函数计算给定数组中每个元素的平方根

它是 Numpy.square() 方法的逆操作。

numpy.sqrt() 语法

numpy.sqrt(arr,
          out=None) 

参数

arr 输入阵列
out 如果给定了 out,结果将存储在 out 中。out 应与 arr 形状相同

返回结果

它返回输入数组中每个元素的平方根数组,即使给定 out 也是如此。

示例代码:numpy.sqrt()

import numpy as np

arr = [1, 9, 25, 49]

arr_sqrt = np.sqrt(arr)

print(arr_sqrt)

输出:

[1. 3. 5. 7.]

示例代码:numpy.sqrt(),参数为 out

import numpy as np

arr = [1, 9, 25, 49]
out_arr = np.zeros(4)

arr_sqrt = np.sqrt(arr, out_arr)

print(out_arr)
print(arr_sqrt)

输出:

[1. 3. 5. 7.]
[1. 3. 5. 7.]

out_arr 的形状与 arr 相同,arr 的平方根保存在其中。而 numpy.sqrt() 方法也返回平方根数组,如上图所示。

如果 outarr 的形状不一样,就会引发 ValueError

import numpy as np

arr = [1, 9, 25, 49]
out_arr = np.zeros((2, 2))

arr_sqrt = np.sqrt(arr, out_arr)

print(out_arr)
print(arr_sqrt)

输出:

Traceback (most recent call last):
  File "C:\Users\jinku\Downloads\test.py", line 6, in <module>
    arr_sqrt = np.sqrt(arr, out_arr)
ValueError: operands could not be broadcast together with shapes (4,) (2,2) 

示例代码:numpy.sqrt() 与负数的关系

import numpy as np

arr = [-1, -9, -25, -49]

arr_sqrt = np.sqrt(arr)

print(arr_sqrt)

输出:

Warning (from warnings module):
  File "..\test.py", line 5
    arr_sqrt = np.sqrt(arr)
RuntimeWarning: invalid value encountered in sqrt
[nan nan nan nan]

当输入是负数时,它会抛出一个 RuntimeWarning,并返回 nan 作为结果。

示例代码:numpy.sqrt() 与复数一起使用

import numpy as np

arr = [3+4j, -5+12j, 8-6j, -15-8j]

arr_sqrt = np.sqrt(arr)

print(arr_sqrt)

输出:

[2.+1.j 2.+3.j 3.-1.j 1.-4.j]

一个复数有两个平方根例如:

numpy sqrt 示例

numpy.sqrt() 方法只返回一个平方根,它有一个正的实数。

comments powered by Disqus