两个数据柱比的TypeError : tan^-1 (tan逆)

Question

我有两个dataframe列"vibration_X“和"vibration_Y".Also --在某些列中几乎没有零值。我想要创建新的列"theta“，它是vibration_Y和vibration_X的tan反比。

以下是我的样本数据：

     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:无法将系列转换为“浮动”类

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