p ython pandas添加一列
时间: 2023-06-05 16:48:10 浏览: 97
python实现在pandas.DataFrame添加一行
可以使用pandas的DataFrame对象的assign()方法来添加一列。例如:
```python
import pandas as pd
# 创建一个DataFrame对象
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# 添加一列C,值为A列和B列的和
df = df.assign(C=df['A'] + df['B'])
print(df)
```
输出结果为:
```
A B C
1 4 5
1 2 5 7
2 3 6 9
```
在assign()方法中,可以使用lambda表达式来定义新列的值。例如:
```python
# 添加一列D,值为A列的平方
df = df.assign(D=lambda x: x['A'] ** 2)
print(df)
```
输出结果为:
```
A B C D
1 4 5 1
1 2 5 7 4
2 3 6 9 9
```
阅读全文