在 Numpy 中返回复数类型数组的逐元素平方根

numpyserver side programmingprogramming更新于 2025/9/3 3:52:17

要逐元素返回数组的非负平方根,请使用 Python Numpy 中的 numpy.sqrt() 方法。返回一个与 x 形状相同的数组,包含 x 中每个元素的正平方根。如果 x 中有任何元素是复数,则返回一个复数数组(并计算负实数的平方根)。如果 x 中的所有元素都是实数,则 y 也是实数,负元素返回 nan。如果提供了 out,则 y 是对它的引用。如果 x 是标量,则 y 也是标量。

out 是存储结果的位置。如果提供,它必须具有输入广播到的形状。如果未提供或为 None,则返回一个新分配的数组。元组(只能作为关键字参数)的长度必须等于输出的数量。

步骤

首先,导入所需的库 −

import numpy as np

使用 array() 方法创建复杂类型的数组 −

arr = np.array([-1j, 1e-15, 27.+0.j, 68.-2.j, 49.+0.j])

显示数组 −

print("数组...
", arr)

获取数组的类型 −

print("
我们的数组类型...
", arr.dtype)

获取数组的维度 −

print("
我们的数组维度...
",arr.ndim)

获取数组的形状 −

print("
我们的数组形状...
",arr.shape)

要按元素返回数组的非负平方根,请使用 Python Numpy 中的 numpy.sqrt() 方法 −

print("
结果...
",np.sqrt(arr))

示例

import numpy as np

# 使用 array() 方法创建复杂类型的数组
arr = np.array([-1j, 1e-15, 27.+0.j, 68.-2.j, 49.+0.j])

# 显示数组
print("数组...
", arr) # 获取数组的类型 print("
我们的数组类型...
", arr.dtype) # 获取数组的维度 print("
我们的数组维度...
",arr.ndim) # 获取数组的形状 print("
我们的数组形状...
",arr.shape) # 要按元素返回数组的非负平方根,请使用 Python Numpy 中的 numpy.sqrt() 方法 print("
结果...
",np.sqrt(arr))

输出

数组...
[-0.0e+00-1.j 1.0e-15+0.j 2.7e+01+0.j 6.8e+01-2.j 4.9e+01+0.j]

我们的数组类型……
complex128

我们的数组维度……
1

我们的数组形状...
(5,)

结果...
[7.07106781e-01-0.70710678j 3.16227766e-08+0.j
5.19615242e+00+0.j 8.24710269e+00-0.1212547j
7.00000000e+00+0.j ]

相关文章