Pandas DataFrame 插入和删除列：使用 insert() 和 drop() 函数

本指南将教你如何在 Pandas DataFrame 对象中插入和删除列。

插入列

使用 insert() 函数可以在指定位置插入一列。

语法：

df.insert(loc, column, value)

• loc：插入列的位置（基于 0 的索引，0 表示第一列）
• column：新列的名称
• value：新列的值

删除列

使用 drop() 函数可以删除指定的列。

语法：

df.drop(columns='column_name', axis=1)

• columns：要删除的列的名称
• axis=1：表示按列删除

代码示例：

import pandas as pd
import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
df = pd.DataFrame(a, columns=['A', 'B', 'C'])

df.insert(1, 'New Column', [10, 20])
print('插入列后的 DataFrame：')
print(df)

df = df.drop('New Column', axis=1)
print('\n删除列后的 DataFrame：')
print(df)

输出：

插入列后的 DataFrame：
   A  New Column  B  C
0  1          10  2  3
1  4          20  5  6

删除列后的 DataFrame：
   A  B  C
0  1  2  3
1  4  5  6

解释：

1. **导入库：**导入 pandas 和 numpy 库。
2. **创建 DataFrame：**创建一个名为 df 的 DataFrame，包含两行三列的数据。
3. **插入列：**使用 insert() 函数在第二列之前插入名为 'New Column' 的新列，并填充值 [10, 20]。
4. **删除列：**使用 drop() 函数删除名为 'New Column' 的列。
5. **打印输出：**打印插入列后和删除列后的 DataFrame。

通过使用 insert() 和 drop() 函数，你可以轻松地对 Pandas DataFrame 对象进行操作，添加和删除所需的列。