pandas增加字段
时间: 2023-11-28 17:46:06 浏览: 161
添加字段 ADDfield
您可以使用 pandas.DataFrame.assign() 方法或 pandas.DataFrame.apply() 方法来增加字段。具体方法如下:
1. 使用 pandas.DataFrame.assign() 方法增加字段:
```python
import pandas as pd
# 创建数据帧
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
# 使用 assign() 方法增加字段
df = df.assign(D=df['A'] + df['B'])
# 打印结果
print(df)
```
输出结果为:
```
A B C D
0 1 4 7 5
1 2 5 8 7
2 3 6 9 9
```
2. 使用 pandas.DataFrame.apply() 方法增加字段:
```python
import pandas as pd
# 创建数据帧
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
# 自定义函数
def sum_columns(row):
return row['A'] + row['B']
# 使用 apply() 方法增加字段
df['D'] = df.apply(sum_columns, axis=1)
# 打印结果
print(df)
```
输出结果为:
```
A B C D
0 1 4 7 5
1 2 5 8 7
2 3 6 9 9
```
阅读全文