DataFrame 选择列
时间: 2023-07-21 13:59:17 浏览: 88
在Python中,可以使用pandas库中的DataFrame对象来添加列。具体步骤如下:
1. 创建一个DataFrame对象,可以使用pandas库中的read_csv()函数从CSV文件中读取数据,也可以手动创建一个DataFrame对象。
2. 使用DataFrame对象的assign()方法来添加新列,该方法接受一个字典作为参数,字典的键为新列的名称,值为新列的数据。
3. 使用DataFrame对象的insert()方法来插入新列,该方法接受三个参数,第一个参数为新列的位置,第二个参数为新列的名称,第三个参数为新列的数据。
例如,以下代码演示了如何在DataFrame对象中添加新列:
```
import pandas as pd
# 创建一个DataFrame对象
data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
'age': [25, 30, 35, 40]}
df = pd.DataFrame(data)
# 使用assign()方法添加新列
df = df.assign(gender=['F', 'M', 'M', 'M'])
# 使用insert()方法插入新列
df.insert(2, 'salary', [50000, 60000, 70000, 80000])
print(df)
```
输出结果为:
```
name age salary gender
Alice 25 50000 F
1 Bob 30 60000 M
2 Charlie 35 70000 M
3 David 40 80000 M
```
阅读全文