Broadcasting errors and debugging in NumPy - Mini Project: Build & Apply

Broadcasting errors and debugging

📖 Scenario: Imagine you are working with sales data for a small store. You have daily sales numbers and want to apply a discount factor to each day's sales to see the adjusted sales. However, sometimes the shapes of your data arrays don't match, causing errors.

🎯 Goal: You will create two numpy arrays representing sales and discount factors. Then you will identify and fix a broadcasting error by adjusting the shape of the discount array. Finally, you will print the corrected adjusted sales.

📋 What You'll Learn

Create a numpy array called sales with values [100, 200, 300]

Create a numpy array called discounts with values [0.9, 0.8]

Create a new array adjusted_sales by multiplying sales and discounts with correct shapes to avoid broadcasting errors

Print the adjusted_sales array

💡 Why This Matters

🌍 Real World

In real data science, you often combine data arrays of different shapes. Understanding broadcasting helps you avoid errors and write efficient code.

💼 Career

Data scientists and analysts frequently manipulate arrays and matrices. Debugging broadcasting errors is a key skill to ensure correct calculations.

1

Create the sales and discounts arrays

Import numpy as np. Create a numpy array called sales with values [100, 200, 300]. Create another numpy array called discounts with values [0.9, 0.8].

NumPy

# Import numpy and create sales and discounts arrays
# Your code here

Hint

Use np.array() to create arrays. Remember to import numpy first.

2

Add a shape adjustment for discounts

Create a new variable called discounts_reshaped by reshaping discounts to have shape (2, 1) using reshape.

NumPy

import numpy as np
sales = np.array([100, 200, 300])
discounts = np.array([0.9, 0.8])
# Reshape discounts to (2, 1)
# Your code here

Hint

Use discounts.reshape((2, 1)) to change the shape.

3

Multiply sales and reshaped discounts to get adjusted sales

Create a new variable called adjusted_sales by multiplying discounts_reshaped and sales. This will use broadcasting correctly.

NumPy

import numpy as np
sales = np.array([100, 200, 300])
discounts = np.array([0.9, 0.8])
discounts_reshaped = discounts.reshape((2, 1))
# Multiply discounts_reshaped and sales to get adjusted_sales
# Your code here

Hint

Use the * operator to multiply arrays. Broadcasting works when shapes align properly.

4

Print the adjusted sales array

Print the adjusted_sales array to see the result.

NumPy

import numpy as np
sales = np.array([100, 200, 300])
discounts = np.array([0.9, 0.8])
discounts_reshaped = discounts.reshape((2, 1))
adjusted_sales = discounts_reshaped * sales
# Print adjusted_sales
# Your code here

Hint

Use print(adjusted_sales) to display the array.

Practice

1. What causes a broadcasting error in NumPy when working with arrays?

easy

A. When arrays contain only zeros

B. When arrays have the same shape

C. When array shapes are not compatible for element-wise operations

D. When arrays are one-dimensional

Solution

1. Step 1: Understand broadcasting rules

   Broadcasting requires arrays to have compatible shapes, meaning dimensions must be equal or one of them is 1.

2. Step 2: Identify cause of error

   If shapes cannot be matched or stretched to match, a broadcasting error occurs.

3. Final Answer:

   When array shapes are not compatible for element-wise operations -> Option C

4. Quick Check:

   Broadcasting error = incompatible shapes [OK]

Hint: Check if array shapes can align or stretch [OK]

Common Mistakes:

• Assuming arrays with same number of elements always broadcast
• Ignoring dimension sizes when broadcasting
• Thinking zeros cause broadcasting errors

2. Which of the following NumPy operations will cause a broadcasting error?

easy

A. np.array([1, 2, 3]) + np.array([[1, 2], [3, 4]])

B. np.array([1, 2, 3]) + np.array([4, 5, 6])

C. np.array([1, 2, 3]) + np.array([[1], [2], [3]])

D. np.array([[1], [2], [3]]) + np.array([4])

Solution

1. Step 1: Check shapes of arrays in each option

   np.array([1, 2, 3]) + np.array([[1, 2], [3, 4]]) adds shape (3,) and (2,2), which are incompatible for broadcasting.

2. Step 2: Confirm broadcasting compatibility

   Other options have compatible shapes: (3,) and (3,1), (3,) and (3,), (3,1) and (1,).

3. Final Answer:

   np.array([1, 2, 3]) + np.array([[1, 2], [3, 4]]) -> Option A

4. Quick Check:

   Shape mismatch causes error [OK]

Hint: Compare array shapes dimension-wise from right [OK]

Common Mistakes:

• Ignoring that (3,) and (2,2) cannot broadcast
• Confusing 1D and 2D shapes
• Assuming all arrays add without error

3. What is the output of this code?

import numpy as np
x = np.array([[1, 2, 3], [4, 5, 6]])
y = np.array([10, 20])
result = x + y
print(result)

medium

A. Broadcasting error

B. [[11 12 13]
[14 15 16]]

C. [[11 12 13]
[24 25 26]]

D. [[11 22 13]
[14 25 16]]

Solution

1. Step 1: Check shapes of x and y

   x shape is (2,3), y shape is (2,). These shapes are not compatible for broadcasting because trailing dimensions differ and y cannot be stretched to match x.

2. Step 2: Predict result of x + y

   Since shapes are incompatible, NumPy raises a broadcasting error.

3. Final Answer:

   Broadcasting error -> Option A

4. Quick Check:

   Shape (2,3) + (2,) = error [OK]

Hint: Match shapes from right; trailing dims must be equal or 1 [OK]

Common Mistakes:

• Assuming (2,) broadcasts to (2,3)
• Expecting element-wise addition without shape check
• Ignoring broadcasting rules for 2D arrays

4. You get a broadcasting error when adding arrays a with shape (3, 1) and b with shape (3,). How can you fix it?

medium

A. Reshape a to (3,) before adding

B. No fix needed; shapes are compatible

C. Transpose a to (1, 3) before adding

D. Reshape b to (3, 1) before adding

Solution

1. Step 1: Analyze shapes

   a (3,1), b (3,): pad b to (1,3). Trailing: 1 vs 3 (1 stretches), leading: 3 vs 1 (1 stretches). Compatible, result shape (3,3).

2. Step 2: No error occurs

   Direct addition works fine.

3. Final Answer:

   No fix needed; shapes are compatible -> Option B

4. Quick Check:

   (3,1) + (3,) works [OK]

Hint: Verify compatibility: pad lower-rank with leading 1s [OK]

Common Mistakes:

• Thinking (3,1) + (3,) errors
• Reshaping when automatic broadcasting works
• Confusing padding position

5. You have a 3D array arr with shape (4, 1, 5) and a 2D array mat with shape (1, 3). You want to add them without error. What is the correct way?

hard

A. Add directly; broadcasting works automatically

B. Reshape arr to (4, 3, 5) before adding

C. Reshape mat to (1, 1, 3) before adding

D. Reshape mat to (1, 3, 1) before adding

Solution

1. Step 1: Check shapes and broadcasting rules

   arr shape: (4,1,5), mat shape: (1,3). To add, shapes must be compatible from right: 5 vs missing, 1 vs 3, 4 vs 1.

2. Step 2: Reshape mat to match arr's dimensions

   Reshape mat to (1,3,1) so shapes become (4,1,5) and (1,3,1), compatible for broadcasting.

3. Final Answer:

   Reshape mat to (1, 3, 1) before adding -> Option D

4. Quick Check:

   Match dimensions by adding missing axis [OK]

Hint: Add missing dimensions with reshape for broadcasting [OK]

Common Mistakes:

• Assuming direct add works without reshape
• Reshaping arr incorrectly
• Confusing dimension order in reshape