5 Best Ways to Rename Columns in a Pandas DataFrame

by

💡 Problem Formulation: When working with Pandas DataFrames, you might encounter scenarios where the column names are not descriptive or suitable for the analyses you intend to perform. For example, suppose you have a DataFrame with columns named ‘A’, ‘B’, and ‘C’, and you want to rename them to ‘Product’, ‘Category’, and ‘Price’ respectively for better readability and understanding. This article will guide you through different methods to achieve such column renaming.

Method 1: Using DataFrame’s rename() Method

The rename() method in Pandas allows you to rename specific columns by passing a dictionary where keys are the current column names and values are the new column names. This method is particularly flexible as it provides the opportunity to rename a subset of columns without changing the entire structure.

♥️ Info: Are you AI curious but you still have to create real impactful projects? Join our official AI builder club on Skool (only $5): SHIP! - One Project Per Month

Here’s an example:

import pandas as pd

# Sample DataFrame
df = pd.DataFrame({
  'A': [1, 2, 3],
  'B': [4, 5, 6],
  'C': [7, 8, 9]
})

# Renaming the columns
df_renamed = df.rename(columns={'A': 'Product', 'B': 'Category', 'C': 'Price'})
print(df_renamed)

Output:

   Product  Category  Price
0        1         4      7
1        2         5      8
2        3         6      9

In the example above, we first create a sample DataFrame with columns ‘A’, ‘B’, and ‘C’. We then use the rename() method, passing a dictionary that maps the old column names to the new ones. Finally, we print the DataFrame with the renamed columns.

Method 2: Assigning to DataFrame.columns Attribute

You can directly assign a new list of column names to the DataFrame.columns attribute. This method is very straightforward and is suitable when you want to rename all the columns at once.

Here’s an example:

import pandas as pd

# Sample DataFrame
df = pd.DataFrame({
  'A': [1, 2, 3],
  'B': [4, 5, 6],
  'C': [7, 8, 9]
})

# Assigning new column names
df.columns = ['Product', 'Category', 'Price']
print(df)

Output:

   Product  Category  Price
0        1         4      7
1        2         5      8
2        3         6      9

In the code snippet above, we replace the current column names by setting the DataFrame.columns attribute with a new list of column names. Note that the length of the list must match the number of columns in the DataFrame.

Method 3: Using the In-Place Renaming Feature

The rename() method can be used with its inplace=True argument to apply the renaming operation directly to the original DataFrame without the need to create a new DataFrame.

Here’s an example:

import pandas as pd

# Sample DataFrame
df = pd.DataFrame({
  'A': [1, 2, 3],
  'B': [4, 5, 6],
  'C': [7, 8, 9]
})

# Renaming the columns in-place
df.rename(columns={'A': 'Product', 'B': 'Category', 'C': 'Price'}, inplace=True)
print(df)

Output:

   Product  Category  Price
0        1         4      7
1        2         5      8
2        3         6      9

By setting inplace=True in the rename() method, we tell pandas to modify the DataFrame in-place. This means that no new DataFrame is returned and instead, the existing DataFrame is altered.

Method 4: Using a List Comprehension for Partial Renaming

If you need to modify only part of the column names, for example, adding a prefix or suffix to each name, you can use a list comprehension to automate the process.

Here’s an example:

import pandas as pd

# Sample DataFrame
df = pd.DataFrame({
  'A': [1, 2, 3],
  'B': [4, 5, 6],
  'C': [7, 8, 9]
})

# Adding a prefix to each column name
df.columns = ['Product_' + col for col in df.columns]
print(df)

Output:

   Product_A  Product_B  Product_C
0          1          4          7
1          2          5          8
2          3          6          9

The list comprehension iterates over each column name in df.columns and prepends the string ‘Product_’ to each one. The resulting list is then assigned back to df.columns.

Bonus One-Liner Method 5: Using the lambda Function

It’s possible to rename DataFrame columns using a lambda function to apply any transformation you wish. This is a flexible one-liner approach that can be helpful for quick and simple column name changes.

Here’s an example:

import pandas as pd

# Sample DataFrame
df = pd.DataFrame({
  'A': [1, 2, 3],
  'B': [4, 5, 6],
  'C': [7, 8, 9]
})

# Using a lambda function to add a suffix to each column name
df.rename(columns=lambda x: x + '_value', inplace=True)
print(df)

Output:

   A_value  B_value  C_value
0        1        4        7
1        2        5        8
2        3        6        9

This one-liner uses the rename() method with a lambda function that takes each column name (represented by x) and adds the suffix ‘_value’. We also use inplace=True to modify the DataFrame directly.

Summary/Discussion

  • Method 1: rename() Method. Highly versatile. Allows selective renaming without affecting untouched columns. Requires the creation of a dictionary mapping which could be verbose for a large number of columns.
  • Method 2: DataFrame.columns Attribute. Very straightforward. Best for renaming all columns at once. Not ideal for partial renaming or when column names are not known beforehand.
  • Method 3: In-Place Renaming. Avoids creating a new DataFrame, saving memory. The original DataFrame is altered, which might not be desirable if you need to retain the original structure.
  • Method 4: List Comprehension for Partial Renaming. Great for systematic renaming patterns. It’s concise but less readable for those not familiar with list comprehensions.
  • Method 5: Using lambda Function. Offers maximum flexibility in a one-liner format. Useful for applying simple transformations to column names.