Renaming columns in Pandas - Time & Space Complexity

Time Complexity: Renaming columns

O(n)

Understanding Time Complexity

We want to understand how the time it takes to rename columns in a pandas DataFrame changes as the number of columns grows.

How does the work increase when we rename more columns?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

df = pd.DataFrame({f'col{i}': range(5) for i in range(100)})

new_names = {f'col{i}': f'new_col{i}' for i in range(100)}
df.rename(columns=new_names, inplace=True)

This code creates a DataFrame with 100 columns and renames all columns using a dictionary mapping.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: pandas goes through each column name to check if it needs to be renamed.
• How many times: Once for each column in the DataFrame.

How Execution Grows With Input

As the number of columns increases, the time to rename grows roughly in direct proportion.

Input Size (n)	Approx. Operations
10	About 10 checks and renames
100	About 100 checks and renames
1000	About 1000 checks and renames

Pattern observation: The work grows linearly as the number of columns grows.

Final Time Complexity

Time Complexity: O(n)

This means the time to rename columns grows in a straight line with the number of columns.

Common Mistake

[X] Wrong: "Renaming columns happens instantly no matter how many columns there are."

[OK] Correct: Even though renaming is fast, pandas must check each column name, so more columns mean more work.

Interview Connect

Understanding how operations scale with data size helps you write efficient code and explain your choices clearly in real projects.

Self-Check

"What if we rename only a few columns instead of all? How would the time complexity change?"

Practice

1. What is the main purpose of renaming columns in a pandas DataFrame?

easy

A. To delete unwanted columns

B. To make the data easier to read and understand

C. To sort the data by column values

D. To change the data type of columns

Solution

1. Step 1: Understand the role of column names

   Column names help identify what data each column holds.

2. Step 2: Purpose of renaming

   Renaming columns improves clarity and makes the data easier to work with.

3. Final Answer:

   To make the data easier to read and understand -> Option B

4. Quick Check:

   Renaming columns = better readability [OK]

Hint: Renaming columns improves clarity and usability [OK]

Common Mistakes:

• Confusing renaming with deleting columns
• Thinking renaming changes data values
• Assuming renaming sorts data

2. Which of the following is the correct syntax to rename columns in a pandas DataFrame df using a dictionary mapping old names to new names?

easy

A. df.rename(columns={'old_name': 'new_name'}, inplace=True)

B. df.rename({'old_name': 'new_name'}, axis=0)

C. df.rename(columns=['old_name', 'new_name'])

D. df.rename(columns='old_name:new_name')

Solution

1. Step 1: Identify correct method parameters

   The rename method requires a dictionary for columns with keys as old names and values as new names.

2. Step 2: Check syntax correctness

   df.rename(columns={'old_name': 'new_name'}, inplace=True) uses columns= with a dictionary and inplace=True, which is correct syntax.

3. Final Answer:

   df.rename(columns={'old_name': 'new_name'}, inplace=True) -> Option A

4. Quick Check:

   Use rename(columns=dict) with inplace=True [OK]

Hint: Use rename(columns={old:new}) with inplace=True to rename [OK]

Common Mistakes:

• Omitting 'columns=' keyword
• Passing list instead of dictionary
• Using axis=0 for column renaming

3. Given the DataFrame df below:

import pandas as pd
df = pd.DataFrame({
  'A': [1, 2],
  'B': [3, 4]
})
df_renamed = df.rename(columns={'A': 'Alpha', 'B': 'Beta'})
print(df_renamed.columns.tolist())

What will be the output?

medium

A. ['Alpha', 'Beta']

B. ['A', 'B']

C. ['Alpha', 'B']

D. Error: rename requires inplace=True

Solution

1. Step 1: Understand rename without inplace

   Calling rename without inplace=True returns a new DataFrame with renamed columns.

2. Step 2: Check the printed columns

   The new DataFrame df_renamed has columns renamed to 'Alpha' and 'Beta'.

3. Final Answer:

   ['Alpha', 'Beta'] -> Option A

4. Quick Check:

   rename() returns new DataFrame with renamed columns [OK]

Hint: rename() returns new DataFrame unless inplace=True [OK]

Common Mistakes:

• Assuming original df columns change without inplace=True
• Expecting error without inplace=True
• Confusing columns list output

4. What is wrong with this code snippet?

df.rename({'old': 'new'}, inplace=True)

medium

A. The 'inplace' parameter is not valid for rename

B. The rename method cannot rename columns

C. The dictionary keys and values should be lists, not strings

D. The dictionary should be passed to the 'columns' parameter

Solution

1. Step 1: Check rename method parameters

   The rename method requires the dictionary to be passed with the keyword 'columns=' to rename columns.

2. Step 2: Identify the error in the code

   Passing the dictionary without 'columns=' causes pandas to treat it as index renaming, not columns.

3. Final Answer:

   The dictionary should be passed to the 'columns' parameter -> Option D

4. Quick Check:

   Use rename(columns=dict) to rename columns [OK]

Hint: Always use columns= when renaming columns [OK]

Common Mistakes:

• Passing dictionary without columns= keyword
• Confusing index renaming with column renaming
• Thinking inplace=False causes error

5. You have a DataFrame df with columns ['name', 'age', 'city']. You want to rename 'name' to 'full_name' and 'city' to 'location' without changing 'age'. Which code correctly renames only these columns and keeps the rest unchanged?

hard

A. df.rename(columns=['full_name', 'age', 'location'], inplace=True)

B. df.columns = ['full_name', 'age', 'location']

C. df.rename(columns={'name': 'full_name', 'city': 'location'}, inplace=True)

D. df.rename({'name': 'full_name', 'city': 'location'}, inplace=True)

Solution

1. Step 1: Understand selective renaming

   Using rename with a dictionary for columns changes only specified columns, leaving others unchanged.

2. Step 2: Check each option

   df.rename(columns={'name': 'full_name', 'city': 'location'}, inplace=True) uses columns= with a dictionary for selective renaming, leaving 'age' unchanged. In contrast, assigning to df.columns replaces all column names and requires the list to match the exact current order. Passing the dictionary directly without columns= attempts to rename the index instead of columns. Using a list instead of a dictionary for columns= will raise a TypeError.

3. Final Answer:

   df.rename(columns={'name': 'full_name', 'city': 'location'}, inplace=True) -> Option C

4. Quick Check:

   Use rename(columns=dict) for selective renaming [OK]

Hint: Use rename(columns=dict) to rename selected columns only [OK]

Common Mistakes:

• Replacing all columns instead of selective renaming
• Omitting columns= keyword
• Using list instead of dictionary for columns