在数据框中的特定位置插入列

在数据分析中，我们常常需要在数据框中添加一列或在现有列的中间插入一列。本文介绍如何在数据框中的特定位置插入列。

插入一列

我们可以使用 insert() 方法在数据框中插入一列，该方法有三个参数：要插入的列的位置、列的名称和列的初始值。

import pandas as pd

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

# 在第二列后插入一列
df.insert(2, 'C', [7, 8, 9])

print(df)

输出:

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

插入多列

要在数据框中插入多列，可以先将要插入的列定义为一个新的数据框，然后使用 concat() 方法将两个数据框连接起来。可以使用 axis 参数指定连接的方向。

import pandas as pd

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

# 定义要插入的两列
df_new = pd.DataFrame({'C': [7, 8, 9], 'D': [10, 11, 12]})

# 在第二列后插入两列
df = pd.concat([df.iloc[:, :2], df_new, df.iloc[:, 2:]], axis=1)

print(df)

输出:

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

插入一列并赋值

要插入一列并为其赋值，可以直接在 insert() 方法中传递一个 Series 对象作为列的值。

import pandas as pd

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

# 在第二列后插入一列并为其赋值
df.insert(2, 'C', pd.Series([7, 8, 9]))

print(df)

输出:

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

Conclusion

以上是在数据框中的特定位置插入列的方法。这些方法可以大大方便我们进行数据分析和处理。