pandas insert 多列

pandas insert 多列

在 pandas 中，我们经常需要对 DataFrame 进行多列的插入操作。虽然 pandas 提供了多种方法来完成这个任务，但是有时候我们需要在指定位置同时插入多列数据，这时候就需要使用 insert 方法了。

什么是 insert 方法

insert 方法是 pandas 中用于在指定位置插入列数据的方法。它可以同时插入多列数据，并且可以指定插入的位置。 insert 方法的基本语法如下：

DataFrame.insert(loc, column, value, allow_duplicates=False)

参数说明：

• loc：插入列的位置，可以是整数或者字符串。如果是整数，则表示插入的位置索引；如果是字符串，则表示插入在指定列名的后面。
• column：要插入的列的名称。
• value：要插入的列的数据，可以是单个值、列表、Series 或者 DataFrame。
• allow_duplicates：是否允许插入重复的列名，默认为 False。

使用示例

接下来，我们通过一个具体的示例来演示如何使用 insert 方法同时插入多列数据。

import pandas as pd

# 创建一个示例 DataFrame
data = {'A': [1, 2, 3],
        'B': [4, 5, 6]}
df = pd.DataFrame(data)
print("原始 DataFrame：")
print(df)

# 要插入的列数据
new_data = {'C': [7, 8, 9],
            'D': [10, 11, 12]}

# 使用 insert 方法插入多列数据
df.insert(loc=1, column='C', value=new_data['C'])
df.insert(loc=2, column='D', value=new_data['D'])
print("\n插入多列后的 DataFrame：")
print(df)

运行上面的代码，输出如下：

原始 DataFrame：
   A  B
0  1  4
1  2  5
2  3  6

插入多列后的 DataFrame：
   A  C   D  B
0  1  7  10  4
1  2  8  11  5
2  3  9  12  6

从输出可以看出，我们成功地使用 insert 方法在指定位置同时插入了两列数据。

总结

本文详细介绍了 pandas 中如何使用 insert 方法插入多列数据。通过 insert 方法，我们可以方便地在指定位置同时插入多列数据，为数据处理和分析提供了更多的灵活性和方便性。