Реализация векторов для ускорения расчета высоты тона (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

Панды Применить

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