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


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

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

步骤

首先,导入所需的库:

import numpy as np

使用array()方法创建一个具有复数类型的数组:

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

显示数组:

print("Array...
", arr)

获取数组的类型:

print("
Our Array type...
", arr.dtype)

获取数组的维度:

print("
Our Array Dimension...
",arr.ndim)

获取数组的形状:

print("
Our Array Shape...
",arr.shape)

要在Python NumPy中逐元素返回数组的非负平方根,请使用numpy.sqrt()方法:

print("
Result...
",np.sqrt(arr))

示例

import numpy as np

# Create an array with complex type using the array() method
arr = np.array([-1j, 1e-15, 27.+0.j, 68.-2.j, 49.+0.j])

# Display the array
print("Array...
", arr) # Get the type of the array print("
Our Array type...
", arr.dtype) # Get the dimensions of the Array print("
Our Array Dimension...
",arr.ndim) # Get the shape of the Array print("
Our Array Shape...
",arr.shape) # To return the non-negative square-root of an array, element-wise, use the numpy.sqrt() method in Python Numpy print("
Result...
",np.sqrt(arr))

输出

Array...
[-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]

Our Array type...
complex128

Our Array Dimension...
1

Our Array Shape...
(5,)

Result...
[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 ]

更新于:2022年2月7日

1K+ 次浏览

开启你的职业生涯

完成课程获得认证

开始学习
广告