melt() for unpivoting in Pandas - Cheat Sheet & Quick Revision

Recall & Review

beginner

What does the melt() function do in pandas?

The melt() function transforms a DataFrame from wide format to long format by unpivoting selected columns into rows.

beginner

In pandas.melt(), what is the role of the id_vars parameter?

id_vars specifies the columns to keep as identifiers (stay fixed) while other columns are unpivoted into rows.

intermediate

How do var_name and value_name parameters affect the output of melt()?

var_name sets the name of the new column that holds the original column names. value_name sets the name of the new column that holds the values from those columns.

beginner

Why would you use melt() in data analysis?

You use melt() to reshape data for easier analysis, especially when you want to convert wide tables into a tidy long format suitable for plotting or grouping.

intermediate

Given a DataFrame with columns ['Name', 'Math', 'Science'], how would melt() transform it?

It would turn the subject columns ('Math', 'Science') into rows under one column (e.g., 'Subject'), with their scores in another column (e.g., 'Score'), keeping 'Name' as the identifier.

What is the default behavior of pandas.melt() if id_vars is not specified?

A An error is raised

B All columns are kept as identifiers

C All columns are unpivoted into rows

D Only the first column is kept as identifier

If id_vars is not set, all columns are unpivoted into two columns: variable and value.

Which parameter in melt() controls the name of the column that holds the original column names?

A value_name

B var_name

C id_vars

D col_name

var_name sets the name of the new column that contains the original column names.

If you want to keep columns 'A' and 'B' fixed and unpivot the rest, how do you specify melt()?

A melt(df, id_vars=['A', 'B'])

B melt(df, var_name=['A', 'B'])

C melt(df, value_name=['A', 'B'])

D melt(df, columns=['A', 'B'])

Use id_vars=['A', 'B'] to keep those columns fixed during unpivoting.

What type of data format does melt() help create?

A Long format

B Hierarchical format

C Nested format

D Wide format

melt() converts data from wide to long format.

Which of these is NOT a typical use case for melt()?

A Preparing data for plotting

B Converting wide tables to long tables

C Tidying data for analysis

D Combining multiple DataFrames vertically

Combining DataFrames vertically is done with functions like concat(), not melt().

Explain how the melt() function works and why it is useful in data analysis.

Think about turning columns into rows to make data easier to analyze.

Describe a real-life example where you would use melt() to reshape your data.

Imagine you have test scores for students in multiple subjects.

Practice

1. What does the pandas.melt() function primarily do to a DataFrame?

easy

A. Filters rows based on a condition.

B. Sorts the DataFrame rows based on column values.

C. Transforms wide format data into long format data by unpivoting columns.

D. Aggregates data by grouping columns.

Solution

1. Step 1: Understand the purpose of melt()

   The melt() function is used to convert wide data (many columns) into long data (fewer columns, more rows).

2. Step 2: Identify the transformation type

   This process is called unpivoting, where columns become rows.

3. Final Answer:

   Transforms wide format data into long format data by unpivoting columns. -> Option C

4. Quick Check:

   melt() = unpivot wide to long [OK]

Hint: melt() turns columns into rows to reshape data [OK]

Common Mistakes:

• Confusing melt() with sorting or filtering functions
• Thinking melt() aggregates data
• Assuming melt() changes row order only

2. Which of the following is the correct syntax to unpivot columns 'A' and 'B' in DataFrame df using melt() while keeping column 'ID' fixed?

easy

A. pd.melt(df, id_vars=['ID'], value_vars=['A', 'B'])

B. pd.melt(df, value_vars=['ID'], id_vars=['A', 'B'])

C. df.melt(id_vars=['A', 'B'], value_vars=['ID'])

D. df.melt(value_vars=['ID'], id_vars=['A', 'B'])

Solution

1. Step 1: Identify correct parameter usage

   id_vars are columns to keep fixed; value_vars are columns to unpivot. Here, 'ID' is fixed, 'A' and 'B' are unpivoted.

2. Step 2: Match syntax with parameters

   pd.melt(df, id_vars=['ID'], value_vars=['A', 'B']) correctly uses pd.melt(df, id_vars=['ID'], value_vars=['A', 'B']). Other options swap or misuse parameters.

3. Final Answer:

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

4. Quick Check:

   id_vars = fixed columns, value_vars = columns to unpivot [OK]

Hint: id_vars = fixed columns; value_vars = columns to unpivot [OK]

Common Mistakes:

• Swapping id_vars and value_vars
• Using DataFrame method without pd prefix incorrectly
• Passing wrong columns to parameters

3. Given the DataFrame df below:

   ID  Jan  Feb
0  1   10   20
1  2   30   40

What is the output of:

pd.melt(df, id_vars=['ID'], var_name='Month', value_name='Sales')

medium

A. DataFrame with columns: Month, Sales only

B. DataFrame with columns: ID, Month, Sales and 4 rows unpivoted from Jan and Feb

C. DataFrame with columns: ID, Jan, Feb unchanged

D. Error due to missing parameters

Solution

1. Step 1: Understand melt() parameters

   id_vars=['ID'] keeps 'ID' fixed. var_name='Month' renames the variable column. value_name='Sales' renames the value column.

2. Step 2: Predict output structure

   Jan and Feb columns become rows under 'Month' with their values under 'Sales'. Result has 4 rows (2 IDs x 2 months).

3. Final Answer:

   DataFrame with columns: ID, Month, Sales and 4 rows unpivoted from Jan and Feb -> Option B

4. Quick Check:

   melt() output = long format with renamed columns [OK]

Hint: melt() creates rows for each value_vars entry per id_vars row [OK]

Common Mistakes:

• Expecting original wide format output
• Ignoring var_name and value_name renaming
• Thinking melt() drops id_vars

4. Consider this code snippet:

df = pd.DataFrame({'ID':[1,2],'A':[10,20],'B':[30,40]})
result = pd.melt(df, id_vars=['A'], value_vars=['ID', 'B'])

What is the main issue with this code?

medium

A. The id_vars column 'A' should be a fixed identifier, but it is a value column here.

B. Syntax error due to wrong parameter names.

C. No issue; code runs correctly and melts as expected.

D. value_vars contains 'ID' which is usually an identifier and should be in id_vars.

Solution

1. Step 1: Analyze id_vars and value_vars roles

   id_vars should be columns to keep fixed (usually identifiers). value_vars are columns to unpivot.

2. Step 2: Check column roles in code

   'ID' is usually an identifier and should be in id_vars, but here it is in value_vars, which is unusual and likely a mistake.

3. Final Answer:

   value_vars contains 'ID' which is usually an identifier and should be in id_vars. -> Option D

4. Quick Check:

   Identifiers belong in id_vars, not value_vars [OK]

Hint: Keep identifiers in id_vars, not in value_vars [OK]

Common Mistakes:

• Swapping id_vars and value_vars
• Assuming melt() ignores column roles
• Thinking code has syntax error

5. You have a DataFrame with columns: ['Date', 'Store', 'Product_A', 'Product_B']. You want to reshape it so each row shows Date, Store, Product (either 'Product_A' or 'Product_B'), and Sales (the value). Which melt() call achieves this?

hard

A. pd.melt(df, id_vars=['Date', 'Store'], var_name='Product', value_name='Sales')

B. pd.melt(df, id_vars=['Product_A', 'Product_B'], var_name='Product', value_name='Sales')

C. pd.melt(df, value_vars=['Date', 'Store'], var_name='Product', value_name='Sales')

D. pd.melt(df, id_vars=['Date'], value_vars=['Store', 'Product_A', 'Product_B'])

Solution

1. Step 1: Identify fixed and unpivot columns

   'Date' and 'Store' are identifiers to keep fixed (id_vars). 'Product_A' and 'Product_B' are value columns to unpivot.

2. Step 2: Use melt() with correct parameters

   Use id_vars=['Date', 'Store'], var_name='Product' to name the unpivoted column, and value_name='Sales' for values.

3. Final Answer:

   pd.melt(df, id_vars=['Date', 'Store'], var_name='Product', value_name='Sales') -> Option A

4. Quick Check:

   Keep identifiers fixed, unpivot product columns [OK]

Hint: Keep identifiers in id_vars; unpivot product columns with var_name [OK]

Common Mistakes:

• Putting product columns in id_vars
• Unpivoting identifier columns
• Not renaming var_name or value_name for clarity