NumPy sqrt() 函数

NumPy sqrt() 函数用于计算输入数组中所有元素的平方根。它计算数组中每个元素 x 的 √x。

此函数可应用于标量、列表或 NumPy 数组，并将返回一个形状相同的数组，其中包含每个输入值的平方根。如果输入的是负数，除非支持复数，否则将返回 NaN(非数字)。

语法

以下是 NumPy sqrt() 函数的语法 -

numpy.sqrt(x, /, out=None, where=True, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj])

参数

此函数接受以下参数 -

• x: 输入数组或标量。该函数计算数组或标量中每个元素的平方根。
• out(可选): 存储结果的位置。如果提供，则必须具有输入广播到的形状。如果未提供或为 None，则返回一个新分配的数组。
• where(可选): 此条件通过输入进行广播。在条件为 True 的位置，将计算结果。否则，结果将保留其原始值。
• casting(可选): 控制可能发生的数据转换类型。默认为"same_kind"。
• order(可选): 控制结果的内存布局顺序。 'C' 表示 C 阶，'F' 表示 Fortran 阶，'A' 表示如果输入均为 F 则为 'F'，否则为 'C'，'K' 表示尽可能匹配输入的布局。
• dtype(可选): 返回数组的类型以及处理元素的累加器的类型。除非指定 dtype，否则默认使用 x 的 dtype。
• subok(可选): 如果为 True，则将传递子类，否则返回的数组将被强制为基类数组。

返回值

此函数返回一个数组，其中每个元素都是输入数组 x 中相应元素的平方根。如果提供了 out，则返回对 out 的引用。

示例:sqrt() 函数的基本用法

在以下示例中，我们使用 sqrt() 函数计算一维数组中每个元素的平方根 -

import numpy as np

# 创建一维数组
arr = np.array([1, 4, 9, 16])

# 对每个元素应用 sqrt
result = np.sqrt(arr)
print(result)

获得的输出将是 -

[1. 2. 3. 4.]

示例:带广播功能的 sqrt() 函数

在本例中，我们演示了如何使用 sqrt() 函数进行广播。我们创建一个二维数组并对其应用 sqrt 运算 -

import numpy as np

# 创建二维数组
arr = np.array([[1, 4, 9], [16, 25, 36]])

# 对每个元素应用 sqrt 运算
result = np.sqrt(arr)
print(result)

这将产生以下结果 -

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

示例:带标量的 sqrt() 函数

在此示例中，我们将 sqrt() 函数应用于标量值 -

import numpy as np

# 标量值
scalar = 25

# 对标量应用 sqrt
result = np.sqrt(scalar)
print(result)

得到的输出为 -

5.0

示例:带负值的 sqrt() 函数(警告)

在此示例中，我们将 sqrt() 函数应用于包含负值的数组。由于实数系统中负数的平方根未定义，因此除非支持复数，否则它将返回 NaN -

import numpy as np

# 创建一个包含负值的一维数组
arr = np.array([-1, -4, -9])

# 对每个元素应用 sqrt(负值会引发警告)
result = np.sqrt(arr)
print(result)

这将产生以下警告 -

/home/cg/root/673acdd6238d1/main.py:7: RuntimeWarning: invalid value encountered in sqrt
  result = np.sqrt(arr)
[nan nan nan]

numpy_arithmetic_operations.html