使用python向dataframe中指定位置添加一列或多列的方法
时间: 2023-12-10 13:41:48 浏览: 100
使用Python向DataFrame中指定位置添加一列或多列的方法
要向DataFrame中添加一列或多列,可以使用以下方法:
1. 使用直接赋值的方式添加一列:
```python
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df['C'] = [7, 8, 9] # 添加一列
df[['D', 'E']] = [[10, 11], [12, 13], [14, 15]] # 添加多列
```
2. 使用insert方法在指定位置插入一列:
```python
df.insert(loc=1, column='F', value=[16, 17, 18])
```
上述代码将在第二列(loc=1)插入一列名为'F'的列,并将其值设置为[16, 17, 18]。
3. 使用assign方法同时添加多列:
```python
df = df.assign(G=[19, 20, 21], H=[22, 23, 24])
```
上述代码将同时添加两列名为'G'和'H'的列,并将它们的值设置为[19, 20, 21]和[22, 23, 24]。
注意:在使用这些方法时,要确保添加的列数与行数相等,否则会出现ValueError。同时,如果使用insert方法插入列时,要确保指定的位置在DataFrame中存在。
阅读全文