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

Question

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

TypeError: only length-1 arrays can be converted to Python scalars

限制是我不能迭代数组来一次添加一个值。

这是一个显示错误的示例：

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（后者仅适用于单个值）。