Broadcasting compatibility check in NumPy - Mini Project: Build & Apply

Broadcasting Compatibility Check

📖 Scenario: Imagine you are working with two sets of numbers representing sales data from two different stores. You want to check if these two sets can be combined using NumPy broadcasting rules to perform element-wise operations easily.

🎯 Goal: You will create two NumPy arrays, set a configuration variable to select one array's shape, check if the two arrays are compatible for broadcasting, and finally print the result.

📋 What You'll Learn

Create two NumPy arrays with exact shapes and values

Create a configuration variable to select which array's shape to check

Use NumPy's broadcasting rules to check compatibility

Print the compatibility result as True or False

💡 Why This Matters

🌍 Real World

Broadcasting is used in data science to perform operations on datasets of different shapes without writing loops, making calculations faster and code simpler.

💼 Career

Understanding broadcasting helps data scientists and analysts efficiently manipulate and combine data arrays, which is essential for data cleaning, feature engineering, and model input preparation.

1

Create two NumPy arrays

Import NumPy as np. Create a NumPy array called store1 with values [[10, 20, 30], [40, 50, 60]] and another array called store2 with values [5, 15, 25].

NumPy

import numpy as np
# Your code here

Hint

Use np.array() to create arrays with the exact values and shapes.

2

Set configuration variable

Create a variable called check_shape and set it to the shape of store1.

NumPy

import numpy as np
store1 = np.array([[10, 20, 30], [40, 50, 60]])
store2 = np.array([5, 15, 25])
# Your code here

Hint

Use the .shape attribute of the array store1 to get its shape.

3

Check broadcasting compatibility

Use np.broadcast_shapes() with the shapes of store1 and store2 inside a try block to check if they can broadcast. Create a variable called compatible and set it to True if broadcasting is possible, otherwise set it to False in the except block.

NumPy

import numpy as np
store1 = np.array([[10, 20, 30], [40, 50, 60]])
store2 = np.array([5, 15, 25])
check_shape = store1.shape
# Your code here

Hint

Use a try-except block to catch ValueError from np.broadcast_shapes().

4

Print the compatibility result

Print the value of the variable compatible.

NumPy

import numpy as np
store1 = np.array([[10, 20, 30], [40, 50, 60]])
store2 = np.array([5, 15, 25])
check_shape = store1.shape
try:
    np.broadcast_shapes(store1.shape, store2.shape)
    compatible = True
except ValueError:
    compatible = False
# Your code here

Hint

Use print(compatible) to show the result.

Practice

1. Which of the following pairs of shapes are compatible for broadcasting in NumPy?

easy

A. (1, 2, 3) and (2, 3, 1)

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

C. (4, 5) and (4, 5, 1)

D. (3, 1) and (1, 4)

Solution

1. Step 1: Understand broadcasting rules

   Broadcasting compares shapes from right to left. Dimensions must be equal or one must be 1.

2. Step 2: Check each pair

   For (3,1) and (1,4), rightmost dims 1 and 4 are compatible (1 can broadcast to 4), left dims 3 and 1 compatible (1 can broadcast to 3). So compatible.

3. Final Answer:

   (3, 1) and (1, 4) are compatible -> Option D

4. Quick Check:

   Broadcast dims right to left, 1 can expand [OK]

Hint: Check shapes from right; 1 or equal dims broadcast [OK]

Common Mistakes:

• Assuming all different shapes are compatible
• Not comparing dimensions from right to left
• Ignoring that 1 can broadcast to any size

2. Which of the following is the correct way to check broadcasting compatibility of shapes (2, 3) and (3,) using NumPy?

easy

A. numpy.broadcast_shapes((2, 3), (3,))

B. numpy.broadcast((2, 3), (3,))

C. numpy.broadcast_shapes([2, 3], 3)

D. numpy.broadcast((2, 3), (3))

Solution

1. Step 1: Identify the correct function

   NumPy provides numpy.broadcast_shapes to check shape compatibility, which takes tuples as shape arguments.

2. Step 2: Check argument types

   numpy.broadcast_shapes((2, 3), (3,)) uses numpy.broadcast_shapes with tuples, which is correct. numpy.broadcast_shapes([2, 3], 3) uses a scalar for the second argument, which is invalid for this function.

3. Final Answer:

   numpy.broadcast_shapes((2, 3), (3,)) -> Option A

4. Quick Check:

   Use numpy.broadcast_shapes with tuples [OK]

Hint: Use numpy.broadcast_shapes with tuples for shape check [OK]

Common Mistakes:

• Using numpy.broadcast instead of numpy.broadcast_shapes
• Passing scalars instead of sequences
• Missing parentheses or wrong argument types

3. What will be the output of the following code?

import numpy as np
shape = np.broadcast_shapes((4, 1, 3), (1, 5, 3))
print(shape)

medium

A. (4, 1, 3)

B. (4, 5, 3)

C. (1, 5, 3)

D. ValueError: shapes not compatible

Solution

1. Step 1: Compare shapes dimension-wise from right

   Rightmost dims: 3 and 3 match. Next dims: 1 and 5, 1 broadcasts to 5. Leftmost dims: 4 and 1, 1 broadcasts to 4.

2. Step 2: Determine broadcasted shape

   Broadcasted shape is (4, 5, 3) combining max of each dimension.

3. Final Answer:

   (4, 5, 3) -> Option B

4. Quick Check:

   Broadcast dims max or 1 expands [OK]

Hint: Broadcast dims take max or 1 expands to max [OK]

Common Mistakes:

• Confusing which dimension broadcasts to which
• Expecting error when shapes are compatible
• Mixing up order of dimensions

4. The following code throws an error. What is the cause?

import numpy as np
np.broadcast_shapes((3, 2), (2, 3))

medium

A. Shapes are not compatible for broadcasting

B. Function broadcast_shapes does not exist

C. Tuples must be of same length

D. Missing import statement

Solution

1. Step 1: Check broadcasting rules for shapes (3, 2) and (2, 3)

   Compare from right: 2 vs 3 are different and neither is 1, so incompatible.

2. Step 2: Confirm error cause

   Since shapes cannot broadcast, numpy.broadcast_shapes raises ValueError.

3. Final Answer:

   Shapes are not compatible for broadcasting -> Option A

4. Quick Check:

   Broadcast dims must be equal or 1 [OK]

Hint: Check dims right to left; mismatch without 1 causes error [OK]

Common Mistakes:

• Thinking function is missing or typo
• Assuming tuples must be same length
• Ignoring broadcasting dimension rules

5. You have arrays with shapes (7, 1, 5) and (1, 3, 1). You want to add them using broadcasting. What will be the resulting shape?

hard

A. (1, 3, 5)

B. (7, 1, 1)

C. (7, 3, 5)

D. Cannot broadcast these shapes

Solution

1. Step 1: Compare shapes from right to left

   Right dims: 5 and 1, 1 broadcasts to 5. Middle dims: 1 and 3, 1 broadcasts to 3. Left dims: 7 and 1, 1 broadcasts to 7.

2. Step 2: Combine broadcasted dimensions

   Resulting shape is (7, 3, 5) taking max of each dimension.

3. Final Answer:

   (7, 3, 5) -> Option C

4. Quick Check:

   Broadcast dims max or 1 expands [OK]

Hint: Broadcast dims max or 1 expands to max [OK]

Common Mistakes:

• Mixing up dimension order
• Assuming shapes must be identical
• Forgetting 1 can broadcast to any size