transform() for group-level operations in Pandas - Mini Project: Build & Apply

Using transform() for Group-Level Operations in pandas

📖 Scenario: You work in a retail company analyzing sales data. You want to understand how each product's sales compare to the average sales of its category.

🎯 Goal: Build a pandas DataFrame with sales data, then use transform() to calculate the average sales per category and add it as a new column.

📋 What You'll Learn

Create a pandas DataFrame named sales_data with columns 'Category' and 'Sales' using the exact data provided.

Create a variable named group_column and set it to the string 'Category'.

Use groupby() on sales_data with group_column and apply transform() with 'mean' to calculate average sales per category, storing the result in a new column 'Average_Sales'.

Print the sales_data DataFrame to display the final result.

💡 Why This Matters

🌍 Real World

Retail analysts often compare individual product sales to category averages to identify strong or weak performers.

💼 Career

Data scientists and analysts use group-level operations like transform() to create features and insights for reports and machine learning.

1

Create the sales data DataFrame

Create a pandas DataFrame called sales_data with two columns: 'Category' and 'Sales'. Use these exact entries: 'Category': ['Electronics', 'Electronics', 'Clothing', 'Clothing', 'Clothing', 'Furniture'], 'Sales': [200, 150, 100, 120, 90, 300].

Pandas

import pandas as pd
# Create the sales_data DataFrame with the exact data
# Your code here

Hint

Use pd.DataFrame with a dictionary containing the two columns and their exact lists.

2

Set the group column variable

Create a variable called group_column and set it to the string 'Category'.

Pandas

import pandas as pd
sales_data = pd.DataFrame({
    'Category': ['Electronics', 'Electronics', 'Clothing', 'Clothing', 'Clothing', 'Furniture'],
    'Sales': [200, 150, 100, 120, 90, 300]
})
# Create the group_column variable with value 'Category'
# Your code here

Hint

Just assign the string 'Category' to the variable group_column.

3

Calculate average sales per category using transform()

Use groupby() on sales_data with group_column and apply transform('mean') on the 'Sales' column. Assign the result to a new column in sales_data called 'Average_Sales'.

Pandas

import pandas as pd
sales_data = pd.DataFrame({
    'Category': ['Electronics', 'Electronics', 'Clothing', 'Clothing', 'Clothing', 'Furniture'],
    'Sales': [200, 150, 100, 120, 90, 300]
})
group_column = 'Category'
# Use groupby and transform to create 'Average_Sales' column
# Your code here

Hint

Use sales_data.groupby(group_column)['Sales'].transform('mean') to get average sales per category and assign it to sales_data['Average_Sales'].

4

Print the final DataFrame

Print the sales_data DataFrame to display the original sales and the new 'Average_Sales' column.

Pandas

import pandas as pd
sales_data = pd.DataFrame({
    'Category': ['Electronics', 'Electronics', 'Clothing', 'Clothing', 'Clothing', 'Furniture'],
    'Sales': [200, 150, 100, 120, 90, 300]
})
group_column = 'Category'
sales_data['Average_Sales'] = sales_data.groupby(group_column)['Sales'].transform('mean')
# Print the sales_data DataFrame
# Your code here

Hint

Use print(sales_data) to show the DataFrame with the new column.

Practice

1. What does the transform() function do when used on a grouped pandas DataFrame?

easy

A. Applies a function to each group and returns a result with the same size as the original data

B. Aggregates each group into a single summary value

C. Filters out groups based on a condition

D. Sorts the groups in ascending order

Solution

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

   transform() applies a function to each group but returns a result with the same number of rows as the original DataFrame.

2. Step 2: Compare with other group operations

   Unlike aggregation which reduces groups to single values, transform() keeps the original data size, allowing new columns with group-level info.

3. Final Answer:

   Applies a function to each group and returns a result with the same size as the original data -> Option A

4. Quick Check:

   transform() = same size output [OK]

Hint: Remember: transform keeps original rows, aggregation reduces them [OK]

Common Mistakes:

• Confusing transform with aggregation
• Thinking transform filters groups
• Assuming transform changes group order

2. Which of the following is the correct syntax to use transform() to calculate the mean of each group in a DataFrame df grouped by column 'Category'?

easy

A. df.groupby('Category').mean().transform()

B. df.groupby('Category').transform('mean')

C. df.transform('mean').groupby('Category')

D. df.groupby('Category').apply(transform('mean'))

Solution

1. Step 1: Identify correct method chaining

   First group by 'Category' then call transform('mean') to apply mean to each group and keep original size.

2. Step 2: Check other options for syntax errors

   df.transform('mean').groupby('Category') calls transform before grouping, which is invalid. df.groupby('Category').mean().transform() calls mean before transform, which is wrong order. df.groupby('Category').apply(transform('mean')) misuses apply with transform.

3. Final Answer:

   df.groupby('Category').transform('mean') -> Option B

4. Quick Check:

   Group first, then transform [OK]

Hint: Group first, then call transform with function name [OK]

Common Mistakes:

• Calling transform before groupby
• Using apply incorrectly with transform
• Calling aggregation before transform

3. Given the DataFrame df below:

  Category  Value
0        A      5
1        A      7
2        B      3
3        B      9
4        B      6

What is the output of df.groupby('Category')['Value'].transform('max')?

medium

A. [6, 6, 9, 9, 9]

B. [5, 7, 3, 6, 9]

C. [7, 5, 9, 3, 6]

D. [7, 7, 9, 9, 9]

Solution

1. Step 1: Group by 'Category' and find max per group

   Group A has values [5,7], max is 7; Group B has [3,9,6], max is 9.

2. Step 2: Apply transform to assign max to each row in group

   Rows in group A get 7, rows in group B get 9, preserving original row order.

3. Final Answer:

   [7, 7, 9, 9, 9] -> Option D

4. Quick Check:

   Max per group repeated for each row [OK]

Hint: transform repeats group result for each row [OK]

Common Mistakes:

• Confusing transform output with aggregation output
• Mixing up max values per group
• Not preserving original row order

4. Consider this code snippet:

df = pd.DataFrame({'Group': ['X', 'X', 'Y'], 'Score': [10, 20, 30]})
df['MeanScore'] = df.groupby('Group')['Score'].transform(mean)

What is the error in this code?

medium

A. The function name 'mean' should be a string: 'mean'

B. Group column name is incorrect

C. transform() cannot be used with mean function

D. DataFrame creation syntax is wrong

Solution

1. Step 1: Check function argument in transform

   The function mean is passed without quotes, but transform expects a string or a callable.

2. Step 2: Correct usage

   Passing 'mean' as a string fixes the error: transform('mean').

3. Final Answer:

   The function name 'mean' should be a string: 'mean' -> Option A

4. Quick Check:

   Function names in transform must be strings or callables [OK]

Hint: Use quotes around function names in transform [OK]

Common Mistakes:

• Passing function name without quotes
• Assuming transform can't use built-in functions
• Misnaming DataFrame columns

5. You have a DataFrame df with columns 'Team' and 'Points'. You want to add a new column 'PointsDiff' that shows the difference between each player's points and the average points of their team. Which code correctly uses transform() to achieve this?

hard

A. df['PointsDiff'] = df.groupby('Team')['Points'].mean() - df['Points']

B. df['PointsDiff'] = df.groupby('Team')['Points'].transform(lambda x: x.mean())

C. df['PointsDiff'] = df['Points'] - df.groupby('Team')['Points'].transform('mean')

D. df['PointsDiff'] = df.groupby('Team')['Points'].apply(lambda x: x.mean())

Solution

1. Step 1: Calculate team average points with transform

   Use transform('mean') to get average points per team for each row, preserving size.

2. Step 2: Subtract team average from each player's points

   Subtract the transformed mean from the original 'Points' column to get difference per player.

3. Step 3: Check other options

   df['PointsDiff'] = df.groupby('Team')['Points'].mean() - df['Points']: df.groupby('Team')['Points'].mean() returns smaller Series, size mismatch. df['PointsDiff'] = df.groupby('Team')['Points'].transform(lambda x: x.mean()): transform(lambda x: x.mean()) sets PointsDiff to repeated group means, not differences. df['PointsDiff'] = df.groupby('Team')['Points'].apply(lambda x: x.mean()): apply(lambda x: x.mean()) returns smaller Series of means, misalignment.

4. Final Answer:

   df['PointsDiff'] = df['Points'] - df.groupby('Team')['Points'].transform('mean') -> Option C

5. Quick Check:

   Subtract group mean using transform for aligned output [OK]

Hint: Subtract transform('mean') result from original column [OK]

Common Mistakes:

• Using apply instead of transform causing misalignment
• Not subtracting in correct order
• Confusing transform with aggregation