Performance tips and vectorization in SciPy - Mini Project: Build & Apply

Performance Tips and Vectorization with SciPy

📖 Scenario: You work as a data analyst. You have a list of numbers representing daily sales. You want to calculate the square root of each sale quickly. Doing this one by one is slow. Using SciPy's vectorized functions can speed this up.

🎯 Goal: Learn how to use SciPy's vectorized functions to calculate square roots of many numbers efficiently.

📋 What You'll Learn

Create a list of daily sales numbers

Import the sqrt function from scipy.special

Use vectorized sqrt to calculate square roots of all sales

Print the resulting array of square roots

💡 Why This Matters

🌍 Real World

Calculating square roots or other math operations on many numbers quickly is common in data analysis and scientific computing.

💼 Career

Knowing how to use vectorized functions from libraries like SciPy helps you write faster and cleaner code, a valuable skill for data scientists and analysts.

1

Create the daily sales data

Create a list called daily_sales with these exact values: 100, 400, 900, 1600, 2500.

SciPy

# Create the list daily_sales with the exact values
# Your code here

Hint

Use square brackets to create a list and separate numbers with commas.

2

Import the sqrt function from scipy

Write an import statement to import sqrt from scipy.special.

SciPy

daily_sales = [100, 400, 900, 1600, 2500]
# Import sqrt from scipy.special
# Your code here

Hint

Use from scipy.special import sqrt to import the square root function.

3

Calculate square roots using vectorized sqrt

Use the sqrt function on daily_sales and save the result in a variable called sales_roots.

SciPy

daily_sales = [100, 400, 900, 1600, 2500]
from scipy.special import sqrt
# Calculate square roots of daily_sales using sqrt and save to sales_roots
# Your code here

Hint

Call sqrt with daily_sales as argument and assign it to sales_roots.

4

Print the square roots

Print the variable sales_roots to see the square roots of the daily sales.

SciPy

daily_sales = [100, 400, 900, 1600, 2500]
from scipy.special import sqrt
sales_roots = sqrt(daily_sales)
# Print sales_roots
# Your code here

Hint

Use print(sales_roots) to display the result.

Practice

1. What is the main benefit of vectorization in SciPy and NumPy?

easy

A. It makes code harder to read but more secure

B. It speeds up calculations by operating on whole arrays at once

C. It requires writing explicit loops for better control

D. It only works with small datasets

Solution

1. Step 1: Understand vectorization concept

   Vectorization means applying operations to entire arrays without explicit loops.

2. Step 2: Identify the main benefit

   This approach speeds up calculations because it uses optimized low-level code.

3. Final Answer:

   It speeds up calculations by operating on whole arrays at once -> Option B

4. Quick Check:

   Vectorization = Faster array operations [OK]

Hint: Vectorization means no loops, faster math on arrays [OK]

Common Mistakes:

• Thinking vectorization requires loops
• Believing vectorization slows code
• Assuming vectorization only works on small data

2. Which of the following is the correct way to add two NumPy arrays a and b element-wise using vectorization?

easy

A. for i in range(len(a)): c[i] = a[i] + b[i]

B. c = np.add(a, b, out=None, where=False)

C. c = a + b

D. c = a.append(b)

Solution

1. Step 1: Review vectorized addition syntax

   NumPy supports element-wise addition directly with c = a + b.

2. Step 2: Check other options

   for i in range(len(a)): c[i] = a[i] + b[i] uses a loop (not vectorized), np.add(a, b, out=None, where=False) has wrong parameters, c = a.append(b) is invalid for arrays.

3. Final Answer:

   c = a + b -> Option C

4. Quick Check:

   Use + for vectorized array addition [OK]

Hint: Use c = a + b for fast element-wise addition [OK]

Common Mistakes:

• Using loops instead of vectorized operators
• Misusing np.add with wrong parameters
• Trying to append arrays for addition

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

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

medium

A. 32

B. array([4, 10, 18])

C. [5, 7, 9]

D. TypeError

Solution

1. Step 1: Understand np.dot with 1D arrays

   np.dot computes the dot product (sum of element-wise products) for 1D arrays.

2. Step 2: Calculate dot product manually

   1*4 + 2*5 + 3*6 = 4 + 10 + 18 = 32

3. Final Answer:

   32 -> Option A

4. Quick Check:

   Dot product sum = 32 [OK]

Hint: np.dot sums element-wise products for 1D arrays [OK]

Common Mistakes:

• Confusing dot product with element-wise multiplication
• Expecting an array instead of a scalar
• Using wrong function for multiplication

4. Identify the error in this vectorized code snippet:

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

medium

A. IndexError because result has no element at index 3

B. TypeError due to multiplying array by integer

C. SyntaxError in array creation

D. No error, prints 6

Solution

1. Step 1: Check array size after multiplication

   Multiplying by 2 keeps array size same: result = [2, 4, 6]

2. Step 2: Accessing index 3

   Index 3 is out of bounds (valid indices: 0,1,2), causing IndexError.

3. Final Answer:

   IndexError because result has no element at index 3 -> Option A

4. Quick Check:

   Array length 3, index 3 invalid [OK]

Hint: Array indices start at 0; max index is length-1 [OK]

Common Mistakes:

• Assuming array length changes after multiplication
• Confusing IndexError with TypeError
• Ignoring zero-based indexing

5. You have a large dataset stored as a NumPy array data. You want to compute the mean of each column efficiently. Which approach is best?

hard

A. Use a for loop to sum each column and divide by number of rows

B. Use np.mean(data) without axis parameter

C. Convert array to list and use Python's built-in sum and len

D. Use np.mean(data, axis=0) to compute means vectorized

Solution

1. Step 1: Understand mean calculation per column

   Mean per column requires averaging along rows (axis=0).

2. Step 2: Identify efficient vectorized method

   np.mean with axis=0 computes column means efficiently without loops.

3. Step 3: Evaluate other options

   Use a for loop to sum each column and divide by number of rows uses slow loops, C converts to list (slow), D computes overall mean, not per column.

4. Final Answer:

   Use np.mean(data, axis=0) to compute means vectorized -> Option D

5. Quick Check:

   Vectorized mean per column = np.mean(data, axis=0) [OK]

Hint: Use np.mean with axis=0 for column-wise mean [OK]

Common Mistakes:

• Using loops instead of vectorized functions
• Forgetting axis parameter in np.mean
• Converting arrays to lists unnecessarily