实施向量以加速音高计算（numpy.arctan2 和 np.sqrt）

Question

正在尝试使用 2 个数据点计算数据的音调信息

使用的函数

def pitch_calculate(x,y,z):
  x_Buff = float(x)
  y_Buff = float(y)
  z_Buff = float(z)
  pitch = np.arctan2((- x_Buff) , np.sqrt(y_Buff * y_Buff + z_Buff * z_Buff)) * 57.3
return pitch

Pandas申请

df['PITCH'] = df.apply(lambda x: pitch_calculate(x['x'], x['y'], x['z'] ), axis=1)

如何为此实现矢量计算？

Answer 1

df['PITCH'] = np.arctan2(-df.x, np.sqrt(df.y * df.y + df.z * df.z)) * 57.3

实施向量以加速音高计算（numpy.arctan2 和 np.sqrt）

Implementing vectors to speed up pitch calculation (numpy.arctan2 and np.sqrt)

python

numpy

vectorization

pandas