Vectorized operations vs loops in Data Analysis Python - Practice Questions

Challenge - 5 Problems

❓ Predict Output

intermediate

Output of vectorized addition vs loop addition

What is the output of the following code comparing vectorized addition with a loop?

Data Analysis Python

import numpy as np
arr = np.array([1, 2, 3, 4])
result_vectorized = arr + 5
result_loop = []
for x in arr:
    result_loop.append(x + 5)
print(result_vectorized)
print(result_loop)

A

[6 7 8 9]
[6, 7, 8, 9]

B

[6 7 8 9]
[5, 6, 7, 8]

C

[6 7 8 9]
[1, 2, 3, 4]

D Error: unsupported operand type(s) for +: 'int' and 'list'

💡 Hint

Think about how numpy arrays handle addition compared to lists.

The numpy array adds 5 to each element directly, producing [6 7 8 9]. The loop appends each element plus 5 to a list, resulting in [6, 7, 8, 9].

❓ data_output

intermediate

Resulting array shape after vectorized multiplication

Given the code below, what is the shape of the resulting array after vectorized multiplication?

Data Analysis Python

import numpy as np
arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([10, 20])
result = arr1 * arr2
print(result.shape)

A (2, 2)

B (1, 2)

C (2,)

D Error: operands could not be broadcast together

💡 Hint

Consider how numpy broadcasts the smaller array to match the larger array's shape.

The array arr2 is broadcast across the rows of arr1, so the result keeps the shape (2, 2).

🔧 Debug

advanced

Identify the error in loop vs vectorized code

What error will the following code produce?

Data Analysis Python

import numpy as np
arr = np.array([1, 2, 3])
result = []
for x in arr:
    result = result + x
print(result)

A SyntaxError: invalid syntax

B TypeError: can only concatenate list (not "int") to list

C [6]

D [1, 2, 3]

💡 Hint

Check how list addition works with integers.

The code tries to add an integer to a list using '+', which causes a TypeError because list concatenation requires another list.

🚀 Application

advanced

Performance difference between vectorized and loop sum

Which statement best describes the performance difference when summing a large array using vectorized numpy sum vs a Python loop?

A Vectorized sum is slower because it creates intermediate arrays.

B Both have the same speed because they do the same operations.

C Vectorized numpy sum is much faster because it uses optimized C code internally.

D Python loop is faster because it processes elements one by one.

💡 Hint

Think about how numpy is implemented under the hood.

Numpy's vectorized operations use optimized compiled code, making them much faster than Python loops for large data.

🧠 Conceptual

expert

Why vectorized operations reduce memory overhead

Why do vectorized operations typically use less memory than equivalent Python loops when processing large datasets?

A Because loops use less memory by processing one element at a time without storing results.

B Because loops automatically compress data during iteration.

C Because vectorized operations store data in multiple copies to speed up access.

D Because vectorized operations avoid creating intermediate Python objects and operate on contiguous memory blocks.

💡 Hint

Consider how numpy arrays store data compared to Python lists.

Vectorized operations work on contiguous blocks of memory and avoid overhead of Python objects, reducing memory use.

Practice

1. What is the main advantage of using vectorized operations over loops in data analysis?

easy

A. They are easier to write for beginners

B. They use less memory by storing fewer variables

C. They perform calculations faster by handling many elements at once

D. They always produce more accurate results

Solution

1. Step 1: Understand vectorized operations

   Vectorized operations process multiple data points simultaneously instead of one by one.

2. Step 2: Compare with loops

   Loops handle data elements one at a time, which is slower for large data.

3. Final Answer:

   They perform calculations faster by handling many elements at once -> Option C

4. Quick Check:

   Vectorized = faster calculations [OK]

Hint: Vectorized means many at once, loops mean one by one [OK]

Common Mistakes:

• Thinking vectorized code always uses less memory
• Believing loops are always easier for beginners
• Assuming vectorized operations improve accuracy

2. Which of the following is the correct syntax to add two numpy arrays a and b using vectorized operations?

easy

A. result = a + b()

B. result = add(a, b)

C. result = a.add(b)

D. result = a + b

Solution

1. Step 1: Recall numpy addition syntax

   Adding two numpy arrays element-wise is done simply by a + b.

2. Step 2: Check other options

   add(a, b) is not a direct function call without importing, a.add(b) is invalid, and a + b() is a syntax error.

3. Final Answer:

   result = a + b -> Option D

4. Quick Check:

   Use + for numpy array addition [OK]

Hint: Use + to add numpy arrays directly [OK]

Common Mistakes:

• Trying to call arrays as functions like b()
• Using methods that don't exist on numpy arrays
• Forgetting to import numpy functions before use

3. What will be the output of this code?

import numpy as np
arr = np.array([1, 2, 3, 4])
result = []
for x in arr:
    result.append(x * 2)
print(result)

medium

A. [1, 2, 3, 4]

B. [2, 4, 6, 8]

C. [3, 6, 9, 12]

D. Error: cannot multiply numpy array element

Solution

1. Step 1: Understand the loop operation

   The loop goes through each element in arr and multiplies it by 2, appending to result.

2. Step 2: Calculate each multiplication

   1*2=2, 2*2=4, 3*2=6, 4*2=8, so result becomes [2, 4, 6, 8].

3. Final Answer:

   [2, 4, 6, 8] -> Option B

4. Quick Check:

   Loop multiply each element by 2 = [2,4,6,8] [OK]

Hint: Loop multiplies each element by 2, collect results [OK]

Common Mistakes:

• Confusing element-wise multiplication with array multiplication
• Expecting original array unchanged
• Thinking numpy arrays can't be iterated

4. Identify the error in this code that tries to double each element of a numpy array using a loop:

import numpy as np
arr = np.array([1, 2, 3])
result = np.array([])
for x in arr:
    result = np.append(result, x * 2)
print(result)

medium

A. Using np.append inside the loop is inefficient and slow

B. np.array([]) creates an integer array, causing type errors

C. Loop variable x is not defined correctly

D. The print statement is missing parentheses

Solution

1. Step 1: Analyze np.append usage in loop

   Using np.append inside a loop creates a new array each time, which is slow and inefficient.

2. Step 2: Check other options

   np.array([]) creates a float array, not integer error; x is correctly defined; print has parentheses.

3. Final Answer:

   Using np.append inside the loop is inefficient and slow -> Option A

4. Quick Check:

   Appending in loop = slow, better vectorize [OK]

Hint: Avoid np.append inside loops; use vectorized ops instead [OK]

Common Mistakes:

• Thinking np.append causes errors instead of inefficiency
• Believing loop variable is undefined
• Ignoring print function syntax

5. You have a large pandas DataFrame df with a column 'values'. Which method is the best to create a new column 'double' that is twice the 'values' column, considering performance?

hard

A. Use df['double'] = df['values'] * 2 for vectorized operation

B. Use a while loop with index to assign doubled values

C. Use df.apply(lambda x: x['values'] * 2, axis=1) to create the column

D. Use a for loop to iterate rows and assign doubled values one by one

Solution

1. Step 1: Compare methods for performance

   Vectorized operations like df['values'] * 2 are fastest and most efficient in pandas.

2. Step 2: Evaluate other options

   Loops and apply are slower because they process rows one by one.

3. Final Answer:

   Use df['double'] = df['values'] * 2 for vectorized operation -> Option A

4. Quick Check:

   Vectorized pandas ops = best performance [OK]

Hint: Multiply whole column at once for speed [OK]

Common Mistakes:

• Using loops instead of vectorized pandas operations
• Choosing apply which is slower than vectorized ops
• Not creating new column properly