datafram变量如何遍历每一行的“A”“B”列相加存入“C”列中
时间: 2023-09-28 10:03:48 浏览: 98
你可以使用 `apply` 方法将一个函数应用到每一行,然后将结果存入新的列中。具体来说,你可以定义一个函数,该函数接受一行作为输入,然后将该行的 `A` 和 `B` 列相加,最后返回相加结果。然后,使用 `apply` 方法将该函数应用到每一行,并将结果存入 `C` 列中。下面是示例代码:
```python
import pandas as pd
# 创建示例数据
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 定义函数将 A 和 B 列相加
def add_columns(row):
return row['A'] + row['B']
# 使用 apply 方法将函数应用到每一行,并将结果存入 C 列中
df['C'] = df.apply(add_columns, axis=1)
# 打印结果
print(df)
```
输出结果:
```
A B C
0 1 4 5
1 2 5 7
2 3 6 9
```
其中,`axis=1` 参数表示将函数应用到每一行。
阅读全文