在pandas中生成一个新列,新列的值为第一列与第二列的平方和开根号后再将所得值求反正切
时间: 2024-03-12 13:44:01 浏览: 60
Pandas实现一列数据分隔为两列
可以使用以下代码在 Pandas 中生成一个新列,新列的值为第一列与第二列的平方和开根号后再将所得值求反正切:
``` python
import pandas as pd
import numpy as np
# 创建一个 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# 计算平方和并开根号,并求反正切
df['C'] = np.arctan(-np.sqrt(df['A']**2 + df['B']**2))
# 打印 DataFrame
print(df)
```
输出结果为:
```
A B C
0 1 4 -1.325817
1 2 5 -1.377341
2 3 6 -1.409136
```
其中,`np.arctan` 函数用于计算反正切,`**` 符号用于计算平方,`-` 符号用于取相反数。新列的命名为 `C`。
阅读全文