Функция sqrt
Функция sqrt вычисляет квадратный корень для каждого элемента переданного массива. Первым параметром функция принимает массив или массивоподобный объект. Вторым параметром можно указать выходной массив для сохранения результата. Работает как с целыми числами, так и с числами с плавающей точкой.
Синтаксис
np.sqrt(arr, [out])
Пример
Давайте вычислим квадратный корень из чисел массива:
import numpy as np
arr = np.array([1, 4, 9, 16, 25])
res = np.sqrt(arr)
print(res)
Результат выполнения кода:
[1. 2. 3. 4. 5.]
Пример
Давайте вычислим квадратный корень из чисел с плавающей точкой:
import numpy as np
arr = np.array([2.25, 5.76, 10.24, 0.36])
res = np.sqrt(arr)
print(res)
Результат выполнения кода:
[1.5 2.4 3.2 0.6]
Пример
Давайте вычислим квадратный корень из отрицательного числа. В результате получим nan (Not a Number):
import numpy as np
arr = np.array([-4, 9, -16, 25])
res = np.sqrt(arr)
print(res)
Результат выполнения кода:
[ nan 3. nan 5.]
Пример
Давайте применим функцию sqrt к двумерному массиву:
import numpy as np
arr = np.array([[1, 4, 9], [16, 25, 36]])
res = np.sqrt(arr)
print(res)
Результат выполнения кода:
[[1. 2. 3.]
[4. 5. 6.]]
Пример
Давайте используем параметр out для сохранения результата в существующий массив:
import numpy as np
arr = np.array([4, 9, 16, 25, 36])
out_arr = np.empty_like(arr)
res = np.sqrt(arr, out=out_arr)
print(res)
Результат выполнения кода:
[2. 3. 4. 5. 6.]
