Element-wise arithmetic in NumPy - Step-by-Step Execution

Concept Flow - Element-wise arithmetic

Start with two arrays

↓

Select operation: +, -, *, /

↓

Apply operation element-wise

↓

Create new array with results

↓

Use or display result

Element-wise arithmetic applies the chosen operation to each pair of elements from two arrays, producing a new array of results.

Execution Sample

NumPy

import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
result = arr1 + arr2
print(result)

Adds two arrays element by element and prints the resulting array.

Execution Table

Step	arr1 element	arr2 element	Operation	Result element
1	1	4	1 + 4	5
2	2	5	2 + 5	7
3	3	6	3 + 6	9
End	-	-	-	All elements processed, result array complete

💡 All elements processed, operation applied element-wise to entire arrays

Variable Tracker

Variable	Start	After 1	After 2	After 3	Final
arr1	[1, 2, 3]	[1, 2, 3]	[1, 2, 3]	[1, 2, 3]	[1, 2, 3]
arr2	[4, 5, 6]	[4, 5, 6]	[4, 5, 6]	[4, 5, 6]	[4, 5, 6]
result	[]	[5]	[5, 7]	[5, 7, 9]	[5, 7, 9]

Key Moments - 2 Insights

Why does the result array have the same length as the input arrays?

Because element-wise arithmetic pairs each element from arr1 with the element at the same position in arr2, producing one result per pair as shown in the execution_table rows 1 to 3.

What happens if the arrays have different lengths?

Numpy will raise an error because it cannot pair elements one-to-one if lengths differ, unlike in this example where both arrays have length 3.

Visual Quiz - 3 Questions

Look at the execution_table, what is the result element at step 2?

A 5

B 9

C 7

D 2

💡 Hint

Check the 'Result element' column at step 2 in the execution_table.

At which step does the result array first contain two elements?

A After step 1

B After step 2

C After step 3

D At the start

💡 Hint

Look at the variable_tracker row for 'result' and see when it reaches length 2.

If arr2 was [4, 5] instead of [4, 5, 6], what would happen?

A Numpy would raise an error due to shape mismatch

B The operation would complete normally with shorter result

C The missing element would be treated as zero

D The operation would ignore extra elements in arr1

💡 Hint

Recall the key_moments explanation about array length mismatch.

Concept Snapshot

Element-wise arithmetic in numpy:
- Applies operations (+, -, *, /) between arrays element by element.
- Both arrays must have the same shape.
- Result is a new array with each element as operation result.
- Example: result = arr1 + arr2 adds corresponding elements.
- Useful for fast, vectorized calculations.

Full Transcript

This visual execution shows how numpy performs element-wise arithmetic. Starting with two arrays of the same length, each element from the first array is paired with the element at the same position in the second array. The chosen operation, here addition, is applied to each pair, producing a new array of results. The execution table traces each step, showing the elements involved and the result computed. The variable tracker shows how the result array grows after each step. Key moments clarify why arrays must be the same length and what happens if they are not. The quiz tests understanding of the step-by-step process and array length requirements. This method allows fast and simple element-wise calculations in data science tasks.

Practice

1. What does element-wise arithmetic mean in numpy?

easy

A. Adding arrays only if they have the same number of elements

B. Applying operations to each pair of elements in arrays individually

C. Multiplying the sum of all elements in two arrays

D. Combining arrays into a single list

Solution

1. Step 1: Understand element-wise operation

   Element-wise arithmetic means applying the operation to each corresponding element in two arrays.

2. Step 2: Compare options

   Only Applying operations to each pair of elements in arrays individually correctly describes this as applying operations to each pair of elements individually.

3. Final Answer:

   Applying operations to each pair of elements in arrays individually -> Option B

4. Quick Check:

   Element-wise = operation on each element pair [OK]

Hint: Think: operation applies to each element pair separately [OK]

Common Mistakes:

• Confusing element-wise with whole-array operations
• Thinking arrays must be identical length only
• Assuming element-wise means concatenation

2. Which of the following is the correct syntax to add two numpy arrays a and b element-wise?

easy

A. np.sum(a, b)

B. a.add(b)

C. a + b

D. a.append(b)

Solution

1. Step 1: Check numpy addition methods

   The + operator performs element-wise addition in numpy.

2. Step 2: Evaluate options

   a.add(b) is invalid because numpy arrays do not have an add method. np.sum(a, b) computes sums, not element-wise addition. a.append(b) is a list method and errors on arrays.

3. Final Answer:

   a + b -> Option C

4. Quick Check:

   + operator adds arrays element-wise [OK]

Hint: Use + for element-wise addition [OK]

Common Mistakes:

• Using methods that don't exist on numpy arrays
• Trying to add arrays with Python list methods
• Confusing element-wise addition with matrix multiplication

3. What is the output of this code?

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

medium

A. [4 10 18]

B. [5 7 9]

C. [1 2 3 4 5 6]

D. Error: arrays must be same shape

Solution

1. Step 1: Understand element-wise multiplication

   Multiplying two numpy arrays multiplies each element pair: 1*4=4, 2*5=10, 3*6=18.

2. Step 2: Calculate output array

   The result is an array: [4, 10, 18].

3. Final Answer:

   [4 10 18] -> Option A

4. Quick Check:

   Element-wise multiply: [1*4, 2*5, 3*6] = [4 10 18] [OK]

Hint: Multiply arrays element-wise: multiply each pair [OK]

Common Mistakes:

• Adding instead of multiplying elements
• Expecting concatenation of arrays
• Thinking multiplication sums elements

4. Identify the error in this code snippet:

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

medium

A. No error, output is [5 7 3]

B. Syntax error in print statement

C. Missing import statement for numpy

D. Arrays have incompatible shapes for element-wise addition

Solution

1. Step 1: Check array shapes

   Array x has shape (3,), y has shape (2,). They are not compatible for element-wise addition.

2. Step 2: Understand numpy broadcasting rules

   Broadcasting requires shapes to be compatible; here they differ and cause an error.

3. Final Answer:

   Arrays have incompatible shapes for element-wise addition -> Option D

4. Quick Check:

   Shape mismatch causes error in element-wise addition [OK]

Hint: Check array shapes before element-wise operations [OK]

Common Mistakes:

• Assuming arrays auto-align despite different lengths
• Ignoring broadcasting rules
• Thinking Python lists behave like numpy arrays

5. Given two arrays:

a = np.array([[1, 2], [3, 4]])
b = np.array([10, 20])

What is the result of a + b and why?

hard

A. [[11 22]
[13 24]] because b is broadcasted to each row of a

B. [[10 20]
[10 20]] because b replaces a

C. [[1 2]
[3 4]] because addition is not possible

D. Error due to shape mismatch

Solution

1. Step 1: Analyze shapes of arrays

   a is shape (2,2), b is shape (2,). Numpy broadcasts b across rows of a.

2. Step 2: Perform element-wise addition with broadcasting

   Add b to each row of a: first row [1+10, 2+20], second row [3+10, 4+20].

3. Final Answer:

   [[11 22] [13 24]] -> Option A

4. Quick Check:

   Broadcasting adds b to each row of a [OK]

Hint: Broadcast smaller array to match larger shape for element-wise ops [OK]

Common Mistakes:

• Expecting error due to different shapes
• Thinking b replaces a instead of broadcasting
• Ignoring numpy broadcasting rules