python - 在 numpy 结构化数组列上执行 math.sqrt

Question

我有一个 numpy 结构化数组。最后一列需要包含基于行中其他值的简单数学方程式的结果。问题是我在尝试计算方程的平方根部分时遇到以下错误:

类型错误:只有长度为 1 的数组可以转换为 Python 标量

限制是我无法迭代数组以一次添加一个值。

这是一个显示错误的例子:

import numpy as np
import math

data = np.random.randint(-100, 100, (1, 6, 4))

data[:,3] = math.sqrt((0-data[:,0])**2 + (0-data[:,1])**2 + (0-data[:,2])**2)

Answer 1

您只需要使用 np.sqrt 而不是 math.sqrt(后者仅适用于单个值)。