pandas insert函数

pandas insert函数

在使用pandas处理数据时，经常需要在DataFrame中插入新的列或者行。pandas提供了insert函数来实现在指定位置插入数据。本文将详细介绍pandas中的insert函数的用法和示例。

insert函数基本用法

insert函数的基本语法如下：

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

参数说明：

• loc：要插入的列的位置，可以是列名或者列索引
• column：要插入的列的名称
• value：要插入的列的值
• allow_duplicates：是否允许出现重复的列名，默认为False

示例代码如下：

import pandas as pd

data = {'A': [1, 2, 3],
        'B': [4, 5, 6]}
df = pd.DataFrame(data)

df.insert(1, 'C', [7, 8, 9])

print(df)

运行结果：

   A  C  B
0  1  7  4
1  2  8  5
2  3  9  6

在DataFrame中插入多列数据

除了插入单列数据，insert函数也可以一次插入多列数据。示例代码如下：

df.insert(2, 'D', [10, 11, 12])
df.insert(3, 'E', [13, 14, 15])

print(df)

运行结果：

   A  C   D   E  B
0  1  7  10  13  4
1  2  8  11  14  5
2  3  9  12  15  6

插入空列

有时候我们需要在DataFrame中插入一列空列，可以先创建一个空的Series，然后插入即可。示例代码如下：

empty_col = pd.Series([], name='F')
df = df.join(empty_col)

print(df)

运行结果：

   A  C   D   E  B   F
0  1  7  10  13  4 NaN
1  2  8  11  14  5 NaN
2  3  9  12  15  6 NaN

使用示例数据演示insert函数的用法

为了更好地说明insert函数的用法，现准备一个示例数据进行演示。示例数据如下：

data = {'Name': ['Alice', 'Bob', 'Charlie'],
        'Age': [25, 30, 35],
        'Gender': ['Female', 'Male', 'Male']}
df = pd.DataFrame(data)

print(df)

运行结果：

      Name  Age  Gender
0    Alice   25  Female
1      Bob   30    Male
2  Charlie   35    Male

现在我们想要在Name和Age列之间插入一个新的列Country，并设置其值为'USA'。示例代码如下：

df.insert(1, 'Country', 'USA')
print(df)

运行结果：

      Name Country  Age  Gender
0    Alice     USA   25  Female
1      Bob     USA   30    Male
2  Charlie     USA   35    Male

通过以上示例，我们可以看到成功在Name和Age列之间插入了新的列Country，并设置了其值为'USA'。

结语

pandas中的insert函数是一个非常方便的函数，可以在DataFrame中灵活地插入新的列或者行，帮助我们更好地处理和分析数据。