Why aggregation matters in NumPy

Introduction

Aggregation helps us find simple summaries from lots of numbers. It makes data easier to understand by showing totals, averages, or counts.

You want to know the average score of a class from all students' scores.

You need the total sales from daily sales numbers to see overall performance.

You want to find the highest temperature recorded in a week.

You want to count how many times a certain event happened in your data.

Syntax

NumPy

numpy.sum(array)
numpy.mean(array)
numpy.max(array)
numpy.min(array)
numpy.count_nonzero(array)

Aggregation functions take an array and return a single number.

You can apply them to the whole array or along specific rows or columns.

Examples

This adds all numbers in the array: 1+2+3+4+5 = 15.

NumPy

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
print(np.sum(arr))

This finds the average: (1+2+3+4+5)/5 = 3.0.

NumPy

print(np.mean(arr))

This finds the biggest number: 5.

NumPy

print(np.max(arr))

This counts how many numbers are greater than 3: 2 (which are 4 and 5).

NumPy

print(np.count_nonzero(arr > 3))

Sample Program

This program shows how aggregation helps summarize sales data. It finds the total sales, average sales per day, and the highest sales each store made on any day.

NumPy

import numpy as np

# Create a 2D array representing daily sales for 3 stores over 4 days
sales = np.array([
    [10, 15, 20, 25],
    [5, 10, 15, 20],
    [8, 12, 18, 22]
])

# Total sales for all stores and days
total_sales = np.sum(sales)

# Average sales per day (across all stores)
avg_sales_per_day = np.mean(sales, axis=0)

# Maximum sales in a single day for each store
max_sales_per_store = np.max(sales, axis=1)

print(f"Total sales: {total_sales}")
print(f"Average sales per day: {avg_sales_per_day}")
print(f"Max sales per store: {max_sales_per_store}")

Important Notes

Aggregation reduces many numbers into a few meaningful values.

Using axis=0 aggregates down columns; axis=1 aggregates across rows.

Aggregation is useful for quick insights and comparisons.

Summary

Aggregation turns many data points into simple summaries.

Common aggregations include sum, mean, max, min, and count.

They help us understand data quickly and clearly.

Practice

1. What is the main purpose of aggregation in data analysis using numpy?

easy

A. To remove missing values from data

B. To create new columns in a dataset

C. To summarize many data points into simple values like sums or averages

D. To sort data in ascending order

Solution

1. Step 1: Understand aggregation meaning

   Aggregation means combining many values into one summary value.

2. Step 2: Identify common aggregation uses

   Common aggregations are sum, mean, max, min, which summarize data quickly.

3. Final Answer:

   To summarize many data points into simple values like sums or averages -> Option C

4. Quick Check:

   Aggregation = summary values [OK]

Hint: Aggregation means summarizing data points simply [OK]

Common Mistakes:

• Confusing aggregation with sorting
• Thinking aggregation creates new columns
• Mixing aggregation with data cleaning

2. Which of the following is the correct syntax to calculate the mean of a numpy array arr?

easy

A. arr.mean()

B. mean(arr)

C. numpy.mean(arr)

D. arr.average()

Solution

1. Step 1: Recall numpy array methods

   Numpy arrays have a method called mean() to calculate average.

2. Step 2: Check syntax correctness

   Calling arr.mean() is correct; mean(arr) is invalid without import, numpy.mean(arr) requires prefix, arr.average() does not exist.

3. Final Answer:

   arr.mean() -> Option A

4. Quick Check:

   Array method for mean = arr.mean() [OK]

Hint: Use array.method() for aggregation in numpy [OK]

Common Mistakes:

• Using mean(arr) without numpy prefix or import
• Using arr.average() which is not a numpy method
• Confusing function and method syntax

3. What is the output of this code?

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
print(arr.sum())

medium

A. 5

B. 15

C. [1 2 3 4 5]

D. Error

Solution

1. Step 1: Understand the array values

   The array contains numbers 1, 2, 3, 4, 5.

2. Step 2: Calculate the sum of array elements

   Sum = 1+2+3+4+5 = 15.

3. Final Answer:

   15 -> Option B

4. Quick Check:

   Sum of 1 to 5 = 15 [OK]

Hint: Sum adds all elements in the array [OK]

Common Mistakes:

• Confusing sum with max or count
• Expecting array output instead of single value
• Syntax errors from missing import

4. Identify the error in this code snippet:

import numpy as np
arr = np.array([1, 2, 3, 4])
result = arr.mean
print(result)

medium

A. mean() method does not exist in numpy

B. Using print instead of return

C. Array should be a list for mean to work

D. Missing parentheses when calling mean method

Solution

1. Step 1: Check how mean is called

   In the code, arr.mean is assigned without parentheses, so it is a method reference, not a value.

2. Step 2: Understand method call syntax

   To get the mean value, arr.mean() must be called with parentheses before assignment.

3. Final Answer:

   Missing parentheses when calling mean method -> Option D

4. Quick Check:

   Method call needs () [OK]

Hint: Call methods with () to get results [OK]

Common Mistakes:

• Assigning method without calling it
• Thinking print causes error
• Believing mean only works on lists

5. You have a 2D numpy array representing sales data for 3 products over 4 days:

import numpy as np
sales = np.array([[10, 20, 30, 40],
                  [5, 15, 25, 35],
                  [2, 4, 6, 8]])

Which code correctly calculates the total sales per product?

hard

A. sales.sum(axis=1)

B. sales.sum(axis=0)

C. sales.mean(axis=1)

D. sales.max(axis=0)

Solution

1. Step 1: Understand array shape and axis meaning

   The array shape is (3 products, 4 days). Axis 0 is down rows (products), axis 1 is across columns (days per product).

2. Step 2: Choose axis to sum per product

   Summing along axis=1 adds sales across days for each product, giving total sales per product.

3. Final Answer:

   sales.sum(axis=1) -> Option A

4. Quick Check:

   Sum per product = sum across columns (axis=1) [OK]

Hint: Sum axis=1 sums rows (per product) in 2D array [OK]

Common Mistakes:

• Using axis=0 sums per day, not per product
• Using mean instead of sum for totals
• Confusing axis numbering in numpy