Element-wise arithmetic in NumPy

Introduction

Element-wise arithmetic lets you do math on each item in a list or array separately. It helps you quickly calculate results for many numbers at once.

Adding sales numbers from two different stores day by day.

Multiplying each pixel value in an image by a brightness factor.

Subtracting temperatures recorded on two different days to find the difference.

Dividing total costs by number of items to find cost per item.

Syntax

NumPy

result = array1 <operator> array2

# where <operator> can be +, -, *, /, etc.

Both arrays should have the same shape or be compatible for broadcasting.

Operations happen on each pair of elements at the same position.

Examples

Adds each element of arr1 to the corresponding element of arr2.

NumPy

import numpy as np

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

Divides each element of arr1 by the corresponding element of arr2.

NumPy

import numpy as np

arr1 = np.array([10, 20, 30])
arr2 = np.array([2, 4, 5])
result = arr1 / arr2
print(result)

Multiplies each element of arr1 by 3 (broadcasting with a single number).

NumPy

import numpy as np

arr1 = np.array([1, 2, 3])
result = arr1 * 3
print(result)

Sample Program

This program adds sales numbers from two stores for each day to find total daily sales.

NumPy

import numpy as np

# Create two arrays representing daily sales from two stores
store1_sales = np.array([100, 150, 200, 250])
store2_sales = np.array([80, 120, 160, 200])

# Calculate total sales per day by adding element-wise
total_sales = store1_sales + store2_sales

print("Store 1 sales:", store1_sales)
print("Store 2 sales:", store2_sales)
print("Total sales per day:", total_sales)

Important Notes

If arrays have different shapes, numpy tries to 'broadcast' smaller arrays to match larger ones.

Be careful with division by zero when doing element-wise division.

Element-wise operations are much faster than looping through elements manually.

Summary

Element-wise arithmetic applies math operations to each pair of elements in arrays.

Arrays must be the same shape or compatible for broadcasting.

This method is useful for quick and easy calculations on many numbers at once.

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