Vectorized operations vs loops in Pandas - Practice Questions

Challenge - 5 Problems

❓ Predict Output

intermediate

Output of vectorized addition vs loop addition

Consider a pandas DataFrame with a column of numbers. What is the output of the following code snippets?

Pandas

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3, 4, 5]})

# Vectorized addition
result_vectorized = df['A'] + 10

# Loop addition
result_loop = []
for x in df['A']:
    result_loop.append(x + 10)

print(result_vectorized.tolist())
print(result_loop)

A [11, 12, 13, 14, 15] and [11, 12, 13, 14, 15]

B [11, 12, 13, 14, 15] and [1, 2, 3, 4, 5]

C [11, 12, 13, 14, 15] and [10, 20, 30, 40, 50]

D [1, 2, 3, 4, 5] and [11, 12, 13, 14, 15]

💡 Hint

Think about how vectorized operations apply the operation to each element automatically.

Both vectorized addition and the loop add 10 to each element in the column, resulting in the same list of values.

❓ data_output

intermediate

Performance difference in execution time

Which of the following statements correctly describes the typical performance difference between vectorized operations and loops in pandas?

A Loops and vectorized operations have the same performance in pandas.

B Vectorized operations are usually slower than loops because they add overhead.

C Vectorized operations are usually faster than loops because they use optimized C code internally.

D Loops are faster because they allow more control over each element.

💡 Hint

Think about how pandas and numpy are implemented under the hood.

Vectorized operations use optimized low-level code, making them faster than explicit Python loops over data.

🔧 Debug

advanced

Identify the error in loop vs vectorized operation

What error will the following code produce and why?

Pandas

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3]})

# Incorrect loop to add 5
result = []
for x in df['A']:
    result.append(x + '5')

print(result)

A ValueError: cannot add string to integer

B TypeError: unsupported operand type(s) for +: 'int' and 'str'

C No error, output: [6, 7, 8]

D SyntaxError: invalid syntax

💡 Hint

Check the data types involved in the addition inside the loop.

The code tries to add an integer and a string, which is not allowed in Python and raises a TypeError.

❓ visualization

advanced

Visualizing speed difference between vectorized and loop operations

You run the following code to compare execution times. Which plot correctly shows the expected result?

Pandas

import pandas as pd
import numpy as np
import time
import matplotlib.pyplot as plt

size = 100000
s = pd.Series(np.arange(size))

start = time.time()
result_vec = s + 1
vec_time = time.time() - start

start = time.time()
result_loop = []
for x in s:
    result_loop.append(x + 1)
loop_time = time.time() - start

plt.bar(['Vectorized', 'Loop'], [vec_time, loop_time])
plt.ylabel('Time in seconds')
plt.title('Execution time comparison')
plt.show()

A Bar plot showing Vectorized time much smaller than Loop time

B Bar plot showing Vectorized time much larger than Loop time

C Bar plot showing Vectorized and Loop times almost equal

D Line plot showing Vectorized time increasing and Loop time decreasing

💡 Hint

Think about how pandas optimizes vectorized operations compared to Python loops.

Vectorized operations run much faster than loops, so the bar for Vectorized time is much smaller.

🧠 Conceptual

expert

Why vectorized operations are preferred in pandas

Which of the following is NOT a reason why vectorized operations are preferred over loops in pandas?

A Vectorized operations reduce the amount of Python code and improve readability.

B Vectorized operations minimize overhead by avoiding explicit Python loops.

C Vectorized operations leverage optimized C code for faster execution.

D Vectorized operations automatically parallelize computations across multiple CPU cores.

💡 Hint

Consider what pandas does internally and what it does not do automatically.

Pandas vectorized operations do not automatically parallelize across CPU cores; parallelism requires extra tools.