python中DataFrame计算生成新一列
时间: 2023-04-09 13:01:02 浏览: 208
pyspark给dataframe增加新的一列的实现示例
可以回答这个问题。在Python中,可以使用DataFrame的apply函数来对每一行进行计算,并生成新的一列。例如,可以使用以下代码来计算两列的和,并生成新的一列:
```python
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df['C'] = df.apply(lambda row: row['A'] + row['B'], axis=1)
print(df)
```
输出结果为:
```
A B C
0 1 4 5
1 2 5 7
2 3 6 9
```
阅读全文