Ошибка типа: tan ^ -1 (обратное значение tan) отношения двух столбцов данных

Question

У меня есть два столбца данных: "vib_X" и "vib__Y". У меня также есть несколько нулевых значений в некоторых столбцах. Я хочу создать новый столбец "theta", который будет иметь отношение обратного загара вибрации_Y и vib__X.

Ниже приведены мои образцы данных:

     vibration_Y  vibration_X
0           10            7
1           10            8
2            9            8
3           10           11
4           13            5
5            3            0
6           12            8
7           12            9
8           11           10
9           10           11

Ниже приведен мой код, который я пробовал и в котором я получаю сообщение об ошибке:

df['theta'] = math.atan(df['vibration_Y']/df['vibration_X'].astype(float))

TypeError: невозможно преобразовать серию в класс 'float'

Answer 1

Я считаю, что вам нужно numpy.arctan:

df['theta'] = np.arctan(df['vibration_Y']/df['vibration_X'])
print (df)
   vibration_Y  vibration_X     theta
0           10            7  0.960070
1           10            8  0.896055
2            9            8  0.844154
3           10           11  0.737815
4           13            5  1.203622
5            3            0  1.570796
6           12            8  0.982794
7           12            9  0.927295
8           11           10  0.832981
9           10           11  0.737815