Why transformation reshapes data for analysis in Data Analysis Python - Performance Analysis

Time Complexity: Why transformation reshapes data for analysis

O(n)

Understanding Time Complexity

When we reshape data, we change its structure to make analysis easier.

We want to know how the time to reshape grows as the data gets bigger.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

data = pd.DataFrame({
    'id': [1, 2, 1, 2],
    'variable': ['A', 'A', 'B', 'B'],
    'value': [10, 20, 30, 40]
})

reshaped = data.pivot(index='id', columns='variable', values='value')

This code reshapes a table from long format to wide format using pivot.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Scanning each row to place values in the new shape.
• How many times: Once for each row in the original data.

How Execution Grows With Input

As the number of rows grows, the time to reshape grows roughly the same way.

Input Size (n)	Approx. Operations
10	About 10 operations
100	About 100 operations
1000	About 1000 operations

Pattern observation: The work grows directly with the number of rows.

Final Time Complexity

Time Complexity: O(n)

This means the time to reshape grows in a straight line as data size grows.

Common Mistake

[X] Wrong: "Reshaping data takes the same time no matter how big the data is."

[OK] Correct: The process must look at each row, so more rows mean more work and more time.

Interview Connect

Understanding how reshaping scales helps you explain data preparation steps clearly and confidently.

Self-Check

"What if we used a groupby aggregation instead of pivot? How would the time complexity change?"

Practice

1. Why do we transform data before analysis?

easy

A. To reshape data for easier comparison and visualization

B. To delete important information

C. To make data harder to understand

D. To increase the file size unnecessarily

Solution

1. Step 1: Understand the purpose of data transformation

   Data transformation changes the shape or format of data to make it easier to analyze.

2. Step 2: Identify the benefit of reshaping data

   Reshaping helps in comparing values side-by-side and creating clear visualizations.

3. Final Answer:

   To reshape data for easier comparison and visualization -> Option A

4. Quick Check:

   Transformation = Easier analysis [OK]

Hint: Think: transformation makes data easier to use [OK]

Common Mistakes:

• Confusing transformation with deleting data
• Believing transformation makes data more complex
• Thinking transformation increases file size

2. Which of the following is the correct way to use pandas melt to reshape data?

easy

A. pd.melt(df, id_vars=['score'], value_vars=['id'])

B. pd.melt(df, id_vars=['id'], value_vars=['score'])

C. pd.melt(df, id_vars='score', value_vars='id')

D. pd.melt(df, id_vars=['score', 'id'])

Solution

1. Step 1: Recall pandas melt syntax

   pandas melt requires id_vars as columns to keep and value_vars as columns to unpivot.

2. Step 2: Check pd.melt(df, id_vars=['id'], value_vars=['score'])

   pd.melt(df, id_vars=['id'], value_vars=['score']) correctly uses id_vars=['id'] and value_vars=['score'], matching the syntax.

3. Final Answer:

   pd.melt(df, id_vars=['id'], value_vars=['score']) -> Option B

4. Quick Check:

   Correct melt syntax = pd.melt(df, id_vars=['id'], value_vars=['score']) [OK]

Hint: id_vars are columns to keep, value_vars to unpivot [OK]

Common Mistakes:

• Swapping id_vars and value_vars
• Passing strings instead of lists
• Omitting value_vars parameter

3. Given this DataFrame:

import pandas as pd
df = pd.DataFrame({
  'id': [1, 2],
  'math': [90, 80],
  'english': [85, 88]
})
reshaped = pd.melt(df, id_vars=['id'], value_vars=['math', 'english'])
print(reshaped)

What is the output?

medium

A. A DataFrame with columns: id, math, english and 2 rows

B. A DataFrame with columns: variable, value and 2 rows

C. A DataFrame with columns: id, variable, value and 4 rows

D. An error because of wrong melt usage

Solution

1. Step 1: Understand the original DataFrame

   It has 2 rows and 3 columns: id, math, english.

2. Step 2: Apply pd.melt with id_vars=['id'] and value_vars=['math', 'english']

   This reshapes math and english columns into two rows per id, creating 4 rows total with columns id, variable, value.

3. Final Answer:

   A DataFrame with columns: id, variable, value and 4 rows -> Option C

4. Quick Check:

   melt reshapes wide to long with 4 rows [OK]

Hint: Melt stacks columns into rows, increasing row count [OK]

Common Mistakes:

• Expecting same number of rows after melt
• Confusing columns after melt
• Thinking melt causes error here

4. What is wrong with this code snippet?

import pandas as pd
df = pd.DataFrame({'id':[1,2],'score':[10,20]})
reshaped = pd.melt(df, id_vars='id', value_vars='score')
print(reshaped)

medium

A. print statement is missing parentheses

B. DataFrame creation syntax is incorrect

C. pd.melt does not exist in pandas

D. id_vars and value_vars should be lists, not strings

Solution

1. Step 1: Check pandas melt parameters

   id_vars and value_vars expect list of column names, not single strings.

2. Step 2: Identify error cause

   Passing 'id' and 'score' as strings causes pandas to treat each character as a column, leading to error or wrong output.

3. Final Answer:

   id_vars and value_vars should be lists, not strings -> Option D

4. Quick Check:

   Use lists for id_vars and value_vars [OK]

Hint: Always pass lists for id_vars and value_vars [OK]

Common Mistakes:

• Passing single strings instead of lists
• Misunderstanding melt parameters
• Assuming melt is not in pandas

5. You have sales data in wide format:

import pandas as pd
df = pd.DataFrame({
  'store': ['A', 'B'],
  'Jan': [100, 150],
  'Feb': [120, 160],
  'Mar': [130, 170]
})

You want to analyze monthly sales trends easily. Which transformation helps and why?

hard

A. Use pd.melt to convert months into one column for easier trend analysis

B. Use df.sort_values() to reorder rows by store

C. Use df.drop() to remove month columns

D. Use df.groupby('store') without reshaping

Solution

1. Step 1: Identify the data format and goal

   Data is wide with months as columns; goal is to analyze trends over months.

2. Step 2: Choose transformation to reshape data

   Using pd.melt converts month columns into a single 'month' column and a 'sales' column, making trend analysis easier.

3. Step 3: Explain why other options don't fit

   Sorting or dropping columns doesn't reshape data for trend analysis; grouping without reshaping keeps data wide.

4. Final Answer:

   Use pd.melt to convert months into one column for easier trend analysis -> Option A

5. Quick Check:

   Wide to long format helps trend analysis [OK]

Hint: Melt wide months into one column for trends [OK]

Common Mistakes:

• Trying to analyze trends without reshaping
• Dropping useful columns
• Confusing sorting with reshaping