Поиск квадратного корня чисел в Numpy

Чтобы найти квадратный корень из списка чисел, вы можете использовать функцию numpy.sqrt() в Python.

Функция sqrt() принимает массив (или список) numpy, вычисляет квадратный корень из элементов в списке и возвращает массив с результатом.

Синтаксис

Синтаксис функции:

output = numpy.sqrt([array_like])

Предоставьте sqrt() массив, подобный аргументу, и он вернет ndarray.

Пример 1: из списка чисел

В этом примере мы инициализируем список чисел и найдем квадратный корень из этих чисел.

import numpy as np

#initialize a list
listA = [4, 16, 9, 1, 25, 49]

#find square root of items in the list
output = np.sqrt(listA)

print(output)

Вывод:

[2. 4. 3. 1. 5. 7.]

Мы предоставили точные квадраты в списке, следовательно, мы получили их квадратные корни без десятичного значения.

Пример 2

В этой программе мы предоставим числа, не являющиеся точными квадратами, и найдем их квадратный корень.

import numpy as np

#initialize a list
listA = [2, 3, 5]

#find square root of items in the list
output = np.sqrt(listA)

print(output)

Вывод:

[1.41421356 1.73205081 2.23606798]

Пример 3: нахождение из комплексных чисел

Вы также можете предоставить комплексные числа в качестве элементов списка для вычисления их квадратных корней.

import numpy as np

#initialize a list
listA = [4+1j, 9+16j]

#find square root of items in the list
output = np.sqrt(listA)

print(output)

Вывод:

[2.01532946+0.24809839j 3.69848346+2.16304875j]

Пример 4: из отрицательных чисел

В этом примере мы предоставим некоторые отрицательные числа, sqrt() выдает RuntimeWarning – просто предупреждение. И возвращает nan (не число) для отрицательного элемента в списке.

import numpy as np

#initialize a list
listA = [-4, 9]

#find square root of items in the list
output = np.sqrt(listA)

print(output)

Вывод:

[nan  3.]