NumPy sqrt() - Квадратный корень из матричных элементов
Модуль Python NumPy используется для работы с многомерными массивами и манипуляциями с матрицами. Мы можем использовать функцию NumPy sqrt(), чтобы получить квадратный корень из элементов матрицы.
Пример Python NumPy sqrt()
import numpy
array_2d = numpy.array([[1, 4], [9, 16]], dtype=numpy.float)
print(array_2d)
array_2d_sqrt = numpy.sqrt(array_2d)
print(array_2d_sqrt)
Выход:
[[ 1. 4.]
[ 9. 16.]]
[[1. 2.]
[3. 4.]]
Давайте посмотрим на другой пример, где элементы матрицы не являются квадратами целых чисел. На этот раз мы будем использовать интерпретатор Python.
>>> import numpy
>>>
>>> array = numpy.array([[1, 3], [5, 7]], dtype=numpy.float)
>>>
>>> print(array)
[[1. 3.]
[5. 7.]]
>>>
>>> array_sqrt = numpy.sqrt(array)
>>>
>>> print(array_sqrt)
[[1. 1.73205081]
[2.23606798 2.64575131]]
>>>
NumPy sqrt() Пример бесконечности
Давайте посмотрим, что происходит, когда у нас есть бесконечность в качестве матричного элемента.
>>> array = numpy.array([1, numpy.inf])
>>>
>>> numpy.sqrt(array)
array([ 1., inf])
>>>
Комплексные числа
>>> array = numpy.array([1 + 2j, -3 + 4j], dtype=numpy.complex)
>>>
>>> numpy.sqrt(array)
array([1.27201965+0.78615138j, 1. +2.j ])
>>>
Отрицательные числа
>>> array = numpy.array([4, -4])
>>>
>>> numpy.sqrt(array)
__main__:1: RuntimeWarning: invalid value encountered in sqrt
array([ 2., nan])
>>>
Квадратный корень матрицы с отрицательными числами вызовет RuntimeWarning, а квадратный корень элемента будет возвращен как nan. Ссылка: документы NumPy