python - 向 pandas 数据框插入多列而不是现有的一列

Question

这个问题已经有答案了:

Adding dummy columns to the original dataframe (2 个回答)

已关闭 3 年前。

我有一个问题 - 如何将多个(例如 3)列插入到 DataFrame 中现有列的位置？换句话说，我有一列包含一些分类值，我使用 one-hot 编码对其进行编码 - 结果，我获得了 3 个新列。现在，我想删除原始列并将结果列插入到其位置(而不是插入到数据框的末尾)。关于如何有效地做到这一点有什么想法吗？我将不胜感激任何帮助。

**df1 - Original datafarme** :

   col1 col2  col3
0   4    A    0.5
1   5    B    0.78
2   6    C    0.55
3   7    A    0.78

**df2 - created one-hot encoding of categorical col2** :

   col2_A col2_B  col2_C
0   1       0       0
1   0       1       0
2   0       0       1
3   1       0       0

如何将df2的列插入到df1中，而不是col2来获取:

**Updated df1**

   col1 col2_A col2_b col2_C  col3
0   4    1      0        0    0.5
1   5    0      1        0    0.78
2   6    0      0        1    0.55
3   7    1      0        0    0.78

Answer 1

使用

df_concat = pd.concat([df1, df2], axis=1)

然后删除 col_2 使用

df_concat.drop(['col_2'], axis = 1)