Common broadcasting patterns in NumPy - Step-by-Step Execution

Concept Flow - Common broadcasting patterns

Start with arrays of different shapes

↓

Compare shapes from right to left

↓

Check dimension compatibility

↓

Broadcast arrays

↓

Perform element-wise operation

↓

Result array

Broadcasting compares array shapes from right to left, checks if dimensions are compatible (equal or 1), then stretches arrays to match shapes for element-wise operations.

Execution Sample

NumPy

import numpy as np
A = np.array([[1,2,3],[4,5,6]])
B = np.array([10,20,30])
C = A + B

Add a 2x3 array A to a 1D array B with 3 elements using broadcasting.

Execution Table

Step	A shape	B shape	Broadcasting Check	Broadcasted Shapes	Operation	Result
1	(2,3)	(3,)	Compare dims: 3 vs 3 (compatible), 2 vs missing (B treated as 1)	(2,3), (2,3)	Add element-wise	[[11,22,33],[14,25,36]]
2	N/A	N/A	No more dims to compare	N/A	Finish	Result shape (2,3)

💡 Broadcasting succeeds because B's shape (3,) is compatible with A's shape (2,3) by stretching B along the first dimension.

Variable Tracker

Variable	Start	After Broadcasting	Final
A	[[1,2,3],[4,5,6]]	Same shape (2,3)	Same shape (2,3)
B	[10,20,30]	Broadcasted to [[10,20,30],[10,20,30]] shape (2,3)	Broadcasted shape (2,3)
C	Undefined	Computed element-wise sum	[[11,22,33],[14,25,36]] shape (2,3)

Key Moments - 2 Insights

Why can a 1D array with shape (3,) be added to a 2D array with shape (2,3)?

Because broadcasting treats the missing dimension in the 1D array as 1, allowing it to stretch along that dimension to match the 2D array shape, as shown in execution_table step 1.

What happens if dimensions are not compatible during broadcasting?

An error occurs because arrays cannot be broadcasted. This is shown in the concept flow where 'No' leads to 'Raise error'.

Visual Quiz - 3 Questions

Look at the execution_table at step 1, what is the broadcasted shape of B?

A (3,)

B (1,3)

C (2,3)

D (2,1)

💡 Hint

Check the 'Broadcasted Shapes' column in execution_table step 1.

At which step does the broadcasting check confirm compatibility?

A Step 1

B Step 2

C No step, error raised

D Before step 1

💡 Hint

Look at the 'Broadcasting Check' column in execution_table.

If B had shape (2,1) instead of (3,), what would be the broadcasted shape when added to A (2,3)?

A (2,1)

B (2,3)

C (3,2)

D Error

💡 Hint

Broadcasting allows dimensions of size 1 to stretch to match the other array's dimension.

Concept Snapshot

Common Broadcasting Patterns in numpy:
- Compare shapes from right to left.
- Dimensions are compatible if equal or one is 1.
- Missing dimensions treated as 1.
- Arrays stretch to match shapes.
- Enables element-wise operations on different shaped arrays.

Full Transcript

Broadcasting in numpy allows arrays of different shapes to be used together in operations. It compares their shapes from right to left. If dimensions are equal or one is 1, they are compatible. Missing dimensions are treated as 1. Arrays are virtually stretched to match shapes, enabling element-wise operations. For example, adding a 2x3 array to a 1D array of length 3 works because the 1D array is stretched along the missing dimension. If dimensions are not compatible, numpy raises an error. This process is key to writing efficient numpy code without manually reshaping arrays.