如何向datafram变量中添加新的列
时间: 2024-02-28 21:54:35 浏览: 78
要向DataFrame变量中添加新的列,你可以使用以下两种方法:
1. 使用下标操作符([])并为新列分配一个列表或Series对象。例如,以下代码向名为df的DataFrame中添加了一个名为“Salary”的新列:
```python
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [25, 30, 35]}
df = pd.DataFrame(data)
salary = [4000, 5000, 6000]
df['Salary'] = salary
print(df)
```
输出:
```
Name Age Salary
0 Alice 25 4000
1 Bob 30 5000
2 Charlie 35 6000
```
2. 使用assign()方法。assign()方法可以用来向DataFrame中添加一个或多个新列。例如,以下代码向名为df的DataFrame中添加了名为“Salary”和“Gender”的两个新列:
```python
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [25, 30, 35]}
df = pd.DataFrame(data)
df = df.assign(Salary=[4000, 5000, 6000], Gender=['F', 'M', 'M'])
print(df)
```
输出:
```
Name Age Salary Gender
0 Alice 25 4000 F
1 Bob 30 5000 M
2 Charlie 35 6000 M
```
在这个例子中,我们将新列的名称和值作为参数传递给assign()方法。方法会返回一个新的DataFrame对象,其中包含原始DataFrame和新添加的列。
阅读全文