pandas 列重命名|极客教程

pandas 列重命名

在数据分析和处理过程中，经常会遇到需要对数据集中的列进行重命名的情况。pandas 是 Python 中一个强大的数据处理库，它提供了许多操作数据的方法，其中重命名列就是其中之一。

在本文中，我们将详细介绍如何使用 pandas 对列进行重命名操作，包括修改单个列名、修改多个列名以及使用字典批量重命名列，帮助读者更加灵活地处理数据。

1. 修改单个列名

首先介绍如何修改数据集中的单个列名。使用 rename 方法可以轻松实现这一操作。

下面是一个示例，假设有一个包含姓名和年龄的数据集 data：

import pandas as pd

data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35]
}

df = pd.DataFrame(data)
print(df)

运行结果如下：

      Name  Age
0    Alice   25
1      Bob   30
2  Charlie   35

现在我们想将列名 ‘Name’ 改为 ‘First Name’：

df.rename(columns={'Name': 'First Name'}, inplace=True)
print(df)

运行结果如下：

  First Name  Age
0      Alice   25
1        Bob   30
2    Charlie   35

可以看到，列名 ‘Name’ 已经成功修改为 ‘First Name’。

2. 修改多个列名

除了修改单个列名，有时候我们也需要同时修改多个列名。这时可以使用字典来指定多个列名的新名称。

假设现在有一个包含学生信息的数据集 student_data，其中包含 ‘ID’、’Name’、’Age’、’Gender’ 等列：

student_data = {
    'ID': [1, 2, 3],
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'Gender': ['F', 'M', 'M']
}

df_student = pd.DataFrame(student_data)
print(df_student)

运行结果如下：

   ID     Name  Age Gender
0   1    Alice   25      F
1   2      Bob   30      M
2   3  Charlie   35      M

现在我们想将 ‘ID’ 改为 ‘Student ID’，将 ‘Gender’ 改为 ‘Sex’：

df_student.rename(columns={'ID': 'Student ID', 'Gender': 'Sex'}, inplace=True)
print(df_student)

运行结果如下：

   Student ID     Name  Age Sex
0           1    Alice   25   F
1           2      Bob   30   M
2           3  Charlie   35   M

可以看到，多个列名成功修改为了新的名称。

3. 使用字典批量重命名列

除了一次性修改多个列名外，有时候我们还需要根据特定的规律批量修改列名。这时可以使用字典来映射原列名和新列名。

假设有一个包含销售数据的数据集 sales_data，列名为 ‘Product_A_Sales’、’Product_B_Sales’、’Product_C_Sales’ 等，我们想要将列名中的 ‘Product_’ 替换为空，即去掉这个前缀：

sales_data = {
    'Product_A_Sales': [100, 200, 150],
    'Product_B_Sales': [120, 220, 170],
    'Product_C_Sales': [90, 180, 140]
}

df_sales = pd.DataFrame(sales_data)
print(df_sales)

运行结果如下：

   Product_A_Sales  Product_B_Sales  Product_C_Sales
0              100              120               90
1              200              220              180
2              150              170              140

现在我们想将列名中的 ‘Product_’ 替换为空：

df_sales.rename(columns=lambda x: x.replace('Product_', ''), inplace=True)
print(df_sales)

运行结果如下：

     A_Sales  B_Sales  C_Sales
0       100      120       90
1       200      220      180
2       150      170      140

可以看到，通过使用 lambda 函数和 replace 方法，成功将列名中的 ‘Product_’ 替换为空。

结语

本文详细介绍了 pandas 中如何对列进行重命名操作，包括修改单个列名、修改多个列名以及使用字典批量重命名列。通过灵活运用这些方法，读者可以更加方便地处理数据集中的列名，提高数据处理的效率和准确性。

pandas 列重命名