python 如何对ndarray 每个变量求平方根?

Posted 2023-03-13

你可以使用numpy模块中的sqrt函数来对ndarray中的每个元素求平方根。下面是一个示例代码：
import numpy as np
# 创建一个ndarray数组
arr = np.array([1, 4, 9, 16, 25])
# 对每个元素求平方根
result = np.sqrt(arr)
# 输出结果
print(result)

这个程序使用numpy模块创建了一个包含5个元素的ndarray数组，并使用sqrt函数对每个元素求平方根。最后，将结果输出到控制台。
如果你想对一个多维的ndarray数组中的每个元素求平方根，可以使用同样的方法。只需确保在调用sqrt函数时，指定要对哪个维度进行操作即可。例如：
import numpy as np

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

# 对每个元素求平方根
result = np.sqrt(arr)

# 输出结果
print(result)

这个程序创建了一个2维ndarray数组，并使用sqrt函数对每个元素求平方根。注意，当sqrt函数应用于多维数组时，默认沿着最后一个维度进行操作。因此，上述示例程序将在每个子数组中的每个元素上应用sqrt函数。如果你想对其他维度进行操作，可以使用axis参数来指定。 参考技术A

本回答被提问者采纳