Scalar and array broadcasting in NumPy - Step-by-Step Execution

Concept Flow - Scalar and array broadcasting

Start with scalar and array

↓

Check shapes compatibility

↓

Broadcast scalar to array shape

↓

Perform element-wise operation

↓

Return result array

Broadcasting lets a scalar act like an array of the same shape to do element-wise operations easily.

Execution Sample

NumPy

import numpy as np
arr = np.array([1, 2, 3])
scalar = 5
result = arr + scalar
print(result)

Add a scalar to each element of an array using broadcasting.

Execution Table

Step	Variable	Value/Shape	Action	Result/Output
1	arr	[1 2 3] (shape (3,))	Define array	[1 2 3]
2	scalar	5 (shape ())	Define scalar	5
3	Broadcast scalar	Broadcast 5 to [5 5 5]	Scalar expanded to array shape	[5 5 5]
4	Addition	arr + broadcasted scalar	Element-wise add	[6 7 8]
5	Print result	[6 7 8]	Output final array	[6 7 8]

💡 All elements added with scalar, operation complete.

Variable Tracker

Variable	Start	After Step 3	After Step 4	Final
arr	[1 2 3]	[1 2 3]	[1 2 3]	[1 2 3]
scalar	5	Broadcasted to [5 5 5]	[5 5 5]	[5 5 5]
result	N/A	N/A	[6 7 8]	[6 7 8]

Key Moments - 3 Insights

Why can we add a scalar to an array directly?

Because numpy broadcasts the scalar to match the array's shape, making element-wise addition possible as shown in step 3 of the execution_table.

Does the scalar change its original value during broadcasting?

No, the scalar value stays the same; broadcasting just treats it as if it were an array of the same shape without copying data, as seen in variable_tracker after step 3.

What happens if the array has multiple dimensions?

The scalar still broadcasts to match the full shape of the array, allowing element-wise operations across all elements, similar to the simple 1D example but extended to more dimensions.

Visual Quiz - 3 Questions

Look at the execution_table at step 3, what is the shape of the broadcasted scalar?

A (3,)

B (1,)

C ()

D (5,)

💡 Hint

Check the 'Value/Shape' column at step 3 in the execution_table.

At which step does the element-wise addition happen?

A Step 2

B Step 3

C Step 4

D Step 5

💡 Hint

Look for the 'Action' describing addition in the execution_table.

If the scalar was replaced by an array [5, 5, 5], how would the broadcast step change?

A Scalar would broadcast to shape (3,)

B No broadcasting needed, direct element-wise addition

C Broadcasting would fail due to shape mismatch

D Scalar would broadcast to shape ()

💡 Hint

Refer to how broadcasting works when both operands have the same shape.

Concept Snapshot

Scalar and array broadcasting in numpy:
- Scalars act like arrays of matching shape
- Enables element-wise operations without explicit loops
- Scalar is virtually expanded to array shape
- Efficient and memory-friendly
- Works with arrays of any dimension

Full Transcript

This visual execution shows how numpy broadcasts a scalar to an array's shape to perform element-wise addition. Starting with an array [1, 2, 3] and a scalar 5, numpy treats the scalar as if it were [5, 5, 5]. Then it adds each element of the array to 5, resulting in [6, 7, 8]. Variables track the scalar and array values before and after broadcasting. Key moments clarify why broadcasting works and that the scalar value itself does not change. The quiz tests understanding of broadcasting shape, operation step, and behavior when replacing scalar with an array.