NumPy sqrt() - 矩阵元素的平方根

Python NumPy 模块用于处理多维数组和矩阵操作。我们可以使用 NumPy sqrt() 函数来获取矩阵元素的平方根。

Python NumPy sqrt() 示例

1import numpy
2
3array_2d = numpy.array([[1, 4], [9, 16]], dtype=numpy.float)
4
5print(array_2d)
6
7array_2d_sqrt = numpy.sqrt(array_2d)
8
9print(array_2d_sqrt)

输出:

1[[ 1. 4.]
2 [ 9. 16.]]
3[[1. 2.]
4 [3. 4.]]

Python Numpy Sqrt Example

让我们看看另一个例子,其中矩阵元素不是整数的方块,这次我们将使用Python解释器。

 1>>> import numpy
 2>>> 
 3>>> array = numpy.array([[1, 3], [5, 7]], dtype=numpy.float)
 4>>> 
 5>>> print(array)
 6[[1. 3.]
 7 [5. 7.]]
 8>>> 
 9>>> array_sqrt = numpy.sqrt(array)
10>>> 
11>>> print(array_sqrt)
12[[1. 1.73205081]
13 [2.23606798 2.64575131]]
14>>>

NumPy sqrt() 无限示例

让我们看看当我们把无限作为矩阵元素时会发生什么。

1>>> array = numpy.array([1, numpy.inf])
2>>> 
3>>> numpy.sqrt(array)
4array([ 1., inf])
5>>>

复杂的数字

1>>> array = numpy.array([1 + 2j, -3 + 4j], dtype=numpy.complex)
2>>> 
3>>> numpy.sqrt(array)
4array([1.27201965+0.78615138j, 1.        +2.j        ])
5>>>

Numpy Sqrt Complex Numbers

负数

1>>> array = numpy.array([4, -4])
2>>> 
3>>> numpy.sqrt(array)
4__main__:1: RuntimeWarning: invalid value encountered in sqrt
5array([ 2., nan])
6>>>

具有负数的矩阵的平方根将投射 RuntimeWarning,元素的平方根将返回为 nan. 参考: NumPy Docs