dataframe 指定位置插入多列
时间: 2023-05-08 20:55:48 浏览: 566
要在DataFrame中指定位置插入多列,可以使用pandas的insert()方法。该方法具有以下参数:
insert(loc, column, value, allow_duplicates=False)
- loc: int
要插入列的位置(索引)。
- column: str or int or tuple
要插入的列名或多个列名的列表或元组。
- value: object or int or float
要插入的值或多个值的列表或数组。长度必须与DataFrame的行数相同。
- allow_duplicates: bool
如果为真,则允许插入重复列名(默认为假)。
例如,如果DataFrame df包含以下数据:
| | A | B | C |
|---|---|---|---|
| 0 | 1 | 2 | 3 |
要在B列之后插入两列D和E,可以使用以下代码:
```
import pandas as pd
df = pd.DataFrame({'A':[1],'B':[2],'C':[3]})
cols_to_insert = ['D', 'E']
values_to_insert = [[4,5]]
df.insert(loc = df.columns.get_loc('B')+1, column = cols_to_insert, value = values_to_insert)
```
结果是:
| | A | B | D | E | C |
|---|---|---|---|---|---|
| 0 | 1 | 2 | 4 | 5 | 3 |
可以看到,D和E列已经成功插入到B列之后。这里使用了get_loc()方法来获取B列的索引位置。values_to_insert值是一个含有一行两列的列表,因为要插入两列新数据。如果需要插入更多列,则需要调整values_to_insert列表的大小和组织方式。
阅读全文