Sales data analysis pattern in Data Analysis Python - Time & Space Complexity

Time Complexity: Sales data analysis pattern

O(p * n)

Understanding Time Complexity

When analyzing sales data, we often run patterns to summarize or find trends. Understanding how long these patterns take helps us work efficiently.

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

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

def total_sales_per_product(sales_df):
    result = {}
    for product in sales_df['product'].unique():
        total = sales_df[sales_df['product'] == product]['amount'].sum()
        result[product] = total
    return result

This code calculates total sales amount for each product by checking all sales entries.

Identify Repeating Operations

• Primary operation: Looping over each unique product and filtering sales data for that product.
• How many times: Once for each unique product, and inside that, scanning the entire sales data to sum amounts.

How Execution Grows With Input

As the number of sales entries and products grow, the time to calculate totals grows faster.

Input Size (n)	Approx. Operations
10 sales entries	About 10 x number of products checks
100 sales entries	About 100 x number of products checks
1000 sales entries	About 1000 x number of products checks

Pattern observation: The total work grows roughly with the number of sales times the number of unique products.

Final Time Complexity

Time Complexity: O(p * n)

This means the time grows with the number of products times the number of sales entries.

Common Mistake

[X] Wrong: "The code runs in linear time because it just loops once."

[OK] Correct: The code loops over products, and for each product, it scans all sales entries, so the work multiplies.

Interview Connect

Understanding how nested loops affect time helps you explain your approach clearly and shows you can think about efficiency in real data tasks.

Self-Check

"What if we used a grouping method like pandas groupby instead? How would the time complexity change?"

Practice

1. What is the main purpose of using groupby() in sales data analysis?

easy

A. To sort sales data alphabetically

B. To delete duplicate sales records

C. To visualize sales trends with charts

D. To summarize sales data by categories like product or date

Solution

1. Step 1: Understand the role of groupby()

   The groupby() function groups data based on one or more columns, such as product or date.

2. Step 2: Identify its use in sales data

   Grouping allows us to summarize sales, like total sales per product or per month.

3. Final Answer:

   To summarize sales data by categories like product or date -> Option D

4. Quick Check:

   groupby() groups data for summary [OK]

Hint: Remember: groupby groups data to summarize by categories [OK]

Common Mistakes:

• Thinking groupby deletes data
• Confusing groupby with sorting
• Assuming groupby creates charts

2. Which of the following is the correct syntax to calculate total sales per product using pandas?

easy

A. df['sales'].groupby('product').sum()

B. df.groupby('product')['sales'].sum()

C. df.sum('sales').groupby('product')

D. df.groupby('sales')['product'].sum()

Solution

1. Step 1: Understand pandas groupby syntax

   The correct syntax groups by 'product' column, then sums the 'sales' column.

2. Step 2: Check each option

   df.groupby('product')['sales'].sum() correctly uses df.groupby('product')['sales'].sum(). Others misuse order or columns.

3. Final Answer:

   df.groupby('product')['sales'].sum() -> Option B

4. Quick Check:

   Group by product, then sum sales [OK]

Hint: Group first, then select column and sum [OK]

Common Mistakes:

• Calling sum() before groupby()
• Grouping by sales instead of product
• Selecting columns after sum()

3. Given the DataFrame df below, what is the output of df.groupby('product')['sales'].sum()?

  product  sales
0   A       100
1   B       200
2   A       150
3   B       100
4   C       300

medium

A. product\nA 250\nB 300\nC 300\nName: sales, dtype: int64

B. product\nA 100\nB 200\nC 300\nName: sales, dtype: int64

C. product\nA 150\nB 100\nC 300\nName: sales, dtype: int64

D. product\nA 350\nB 300\nC 300\nName: sales, dtype: int64

Solution

1. Step 1: Group sales by product

   Sum sales for each product: A = 100+150=250, B = 200+100=300, C = 300.

2. Step 2: Write the output as pandas Series

   The result shows sums indexed by product with name 'sales'.

3. Final Answer:

   product\nA 250\nB 300\nC 300\nName: sales, dtype: int64 -> Option A

4. Quick Check:

   Sum sales per product = 250, 300, 300 [OK]

Hint: Add sales values grouped by product [OK]

Common Mistakes:

• Adding sales incorrectly
• Confusing product labels
• Ignoring grouping before sum

4. The following code is intended to find the average sales per month, but it raises an error. What is the error?

df.groupby('month').mean('sales')

medium

A. mean() does not take any arguments; column selection must be before mean()

B. groupby() requires a list, not a string

C. mean() should be replaced with sum()

D. The DataFrame must be sorted before groupby()

Solution

1. Step 1: Check mean() function usage

   The mean() method does not accept column names as arguments.

2. Step 2: Correct usage for column selection

   Select the 'sales' column before calling mean(): df.groupby('month')['sales'].mean().

3. Final Answer:

   mean() does not take any arguments; column selection must be before mean() -> Option A

4. Quick Check:

   mean() no args; select column first [OK]

Hint: Select column before aggregation function [OK]

Common Mistakes:

• Passing column name inside mean()
• Using groupby with wrong argument type
• Thinking sorting is required before groupby

5. You have sales data with columns product, month, and sales. How do you find the top-selling product for each month?

hard

A. Use df.groupby('month')['sales'].max() without product info

B. Use df.groupby('product')['sales'].max() and sort by month

C. Use df.groupby(['month', 'product'])['sales'].sum().unstack().idxmax(axis=1)

D. Use df.sort_values('sales').groupby('month').first() without aggregation

Solution

1. Step 1: Group sales by month and product

   Sum sales for each product in each month using groupby(['month', 'product'])['sales'].sum().

2. Step 2: Reshape and find top product per month

   Use unstack() to make products columns, then idxmax(axis=1) to find product with max sales per month.

3. Final Answer:

   Use df.groupby(['month', 'product'])['sales'].sum().unstack().idxmax(axis=1) -> Option C

4. Quick Check:

   Group by month and product, then find max product [OK]

Hint: Group by month and product, then find max sales product [OK]

Common Mistakes:

• Ignoring product when finding max sales
• Using max() without grouping by product
• Sorting without aggregation