Aggregation functions (sum, mean, count) in Data Analysis Python - Step-by-Step Execution

Concept Flow - Aggregation functions (sum, mean, count)

Start with data

↓

Choose aggregation function

↓

Apply function to data column

↓

Calculate result

↓

Return single summary value

Aggregation functions take a list of numbers and return one summary number like total, average, or count.

Execution Sample

Data Analysis Python

import pandas as pd

data = pd.DataFrame({'values': [10, 20, 30, 40]})

sum_val = data['values'].sum()
mean_val = data['values'].mean()
count_val = data['values'].count()

This code calculates the sum, mean, and count of numbers in the 'values' column.

Execution Table

Step	Action	Data Column	Function Applied	Result
1	Start with data	[10, 20, 30, 40]	None	N/A
2	Apply sum()	[10, 20, 30, 40]	sum	100
3	Apply mean()	[10, 20, 30, 40]	mean	25.0
4	Apply count()	[10, 20, 30, 40]	count	4
5	End	N/A	N/A	All aggregations done

💡 All aggregation functions applied and results returned.

Variable Tracker

Variable	Start	After sum	After mean	After count
sum_val	N/A	100	100	100
mean_val	N/A	N/A	25.0	25.0
count_val	N/A	N/A	N/A	4

Key Moments - 2 Insights

Why does sum_val stay the same after mean and count calculations?

Because sum_val is calculated once at step 2 and does not change afterward, as shown in execution_table rows 2-5.

What does count() count exactly in the data column?

count() counts the number of non-missing values in the column, here it counts 4 values as shown in execution_table row 4.

Visual Quiz - 3 Questions

Look at the execution table, what is the result of mean() at step 3?

A 100

B 25.0

C 4

D 10

💡 Hint

Check the 'Result' column at step 3 in the execution_table.

At which step does the count() function get applied?

A Step 4

B Step 3

C Step 2

D Step 5

💡 Hint

Look at the 'Function Applied' column in the execution_table.

If the data had one missing value, how would count_val change?

A It would stay 4

B It would become 5

C It would become 3

D It would become 0

💡 Hint

count() counts only non-missing values, so one missing reduces count by one.

Concept Snapshot

Aggregation functions summarize data columns.
sum() adds all numbers.
mean() finds average.
count() counts non-missing values.
Use them to get quick insights from data.

Full Transcript

Aggregation functions like sum, mean, and count take a list of numbers and return a single summary number. For example, sum adds all numbers, mean calculates the average, and count counts how many values are present. In the example, we start with a data column containing [10, 20, 30, 40]. Applying sum() gives 100, mean() gives 25.0, and count() gives 4. These results are stored in variables sum_val, mean_val, and count_val respectively. Each variable is calculated once and does not change afterward. Count counts only non-missing values, so if there were missing data, count would be less. These functions help quickly understand data by summarizing it.

Practice

1. What does the sum() function do when applied to a column of numbers in a DataFrame?

easy

A. Counts how many numbers are in the column

B. Finds the middle value of the numbers

C. Adds all the numbers together to get a total

D. Sorts the numbers from smallest to largest

Solution

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

   The sum() function adds all values in a column to get their total.

2. Step 2: Compare with other functions

   mean() finds average, count() counts entries, so they do different things.

3. Final Answer:

   Adds all the numbers together to get a total -> Option C

4. Quick Check:

   sum() = total of values [OK]

Hint: Sum adds values, mean averages, count counts entries [OK]

Common Mistakes:

• Confusing sum() with mean()
• Thinking sum() counts entries
• Assuming sum() sorts data

2. Which of the following is the correct way to calculate the average of a DataFrame column named sales using pandas?

easy

A. df.sales.sum() / df.sales.count()

B. df.mean('sales')

C. mean(df.sales)

D. df['sales'].mean()

Solution

1. Step 1: Identify correct pandas syntax for mean()

   In pandas, to get the mean of a column, use df['column'].mean().

2. Step 2: Check other options

   df.mean('sales') is invalid syntax; mean(df.sales) is not pandas method; df.sales.sum() / df.sales.count() works but is longer and less direct.

3. Final Answer:

   df['sales'].mean() -> Option D

4. Quick Check:

   Use df['column'].mean() for average [OK]

Hint: Use df['col'].mean() to get average quickly [OK]

Common Mistakes:

• Using mean() as a standalone function
• Passing column name as argument to df.mean()
• Dividing sum by count manually instead of mean()

3. Given the DataFrame df below:

import pandas as pd
data = {'score': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

What is the output of df['score'].count()?

medium

A. 5

B. 150

C. 30

D. Error

Solution

1. Step 1: Understand what count() does on a column

   count() counts the number of non-missing entries in the column. Here, all 5 values are present.

2. Step 2: Count the entries in 'score'

   There are 5 numbers: 10, 20, 30, 40, 50, so count() returns 5.

3. Final Answer:

   5 -> Option A

4. Quick Check:

   count() = number of entries [OK]

Hint: count() returns number of non-empty entries [OK]

Common Mistakes:

• Confusing count() with sum()
• Expecting count() to return total value
• Thinking count() returns average

4. Consider this code snippet:

import pandas as pd
data = {'values': [5, None, 15, 20]}
df = pd.DataFrame(data)
result = df['values'].sum()

What is the value of result?

medium

A. 40

B. None

C. Error

D. 20

Solution

1. Step 1: Understand how sum() handles None values

   In pandas, sum() ignores None (missing) values and sums only the numbers.

2. Step 2: Calculate sum ignoring None

   Values are 5, None, 15, 20. Sum = 5 + 15 + 20 = 40.

3. Final Answer:

   40 -> Option A

4. Quick Check:

   sum() skips None and adds numbers [OK]

Hint: sum() ignores None/null values automatically [OK]

Common Mistakes:

• Expecting sum() to return None if any missing
• Getting error due to None value
• Counting None as zero explicitly

5. You have a DataFrame df with a column sales containing daily sales numbers. You want to find the average sales but only for days where sales were above zero. Which code correctly calculates this?

hard

A. df['sales'].mean()

B. df[df['sales'] > 0]['sales'].mean()

C. df['sales'].sum() / df['sales'].count()

D. df['sales'].count() / df['sales'].sum()

Solution

1. Step 1: Filter rows where sales > 0

   Use df[df['sales'] > 0] to keep only positive sales days.

2. Step 2: Calculate mean of filtered sales

   Apply ['sales'].mean() on filtered data to get average of positive sales only.

3. Final Answer:

   df[df['sales'] > 0]['sales'].mean() -> Option B

4. Quick Check:

   Filter then mean() for conditional average [OK]

Hint: Filter data first, then apply mean() for conditional average [OK]

Common Mistakes:

• Using mean() without filtering
• Dividing sum by count without filtering
• Swapping sum and count in division