Broadcasting errors and debugging in NumPy - Cheat Sheet & Quick Revision

Recall & Review

beginner

What is broadcasting in NumPy?

Broadcasting is a way NumPy performs operations on arrays of different shapes by automatically expanding the smaller array to match the shape of the larger one without copying data.

beginner

What causes a broadcasting error in NumPy?

A broadcasting error happens when two arrays have incompatible shapes that cannot be aligned according to broadcasting rules, so NumPy cannot perform element-wise operations.

beginner

How can you check the shape of a NumPy array?

You can check the shape of a NumPy array using the `.shape` attribute, which returns a tuple showing the size of the array in each dimension.

intermediate

What is a common debugging step when facing a broadcasting error?

A common step is to print the shapes of the arrays involved and compare them to see why they cannot be broadcast together, then reshape or expand dimensions as needed.

intermediate

Explain the broadcasting rule for two arrays with shapes (3, 1) and (1, 4).

NumPy compares shapes from right to left: 1 and 4 are compatible because 1 can be broadcast to 4; 3 and 1 are compatible because 1 can be broadcast to 3. The result shape is (3, 4).

What happens if you try to add arrays with shapes (2, 3) and (3, 2) in NumPy?

A They broadcast successfully to shape (3, 2)

B Broadcasting error occurs

C They broadcast successfully to shape (2, 3)

D Result is a scalar

Shapes (2, 3) and (3, 2) are not compatible for broadcasting because neither dimension matches or is 1.

Which attribute helps you find the shape of a NumPy array?

A .shape

B .size

C .length

D .dim

The .shape attribute returns the dimensions of the array as a tuple.

If an array has shape (5, 1) and another has shape (1, 4), what is the shape after broadcasting?

A (1, 1)

B (5, 1)

C (5, 4)

D (1, 4)

Dimensions with 1 can be broadcast to match the other dimension, resulting in (5, 4).

What is a good first step to debug a broadcasting error?

A Print array shapes

B Ignore the error

C Use a for loop

D Restart Python

Printing shapes helps identify why arrays cannot be broadcast together.

Which of these shapes can be broadcast together?

A (3, 2) and (2, 3)

B (2, 2) and (3, 3)

C (1, 3) and (2, 2)

D (4, 1) and (1, 5)

Shapes (4,1) and (1,5) broadcast to (4,5) because 1 can be stretched to match the other dimension.

Describe the main rules NumPy uses to decide if two arrays can be broadcast together.

Think about how NumPy stretches arrays to match shapes.

Explain how you would debug a broadcasting error in your code.

Start by printing array shapes before the operation.

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