Vectorized operations vs loops in Pandas - Performance Comparison

Time Complexity: Vectorized operations vs loops

O(n)

Understanding Time Complexity

We want to see how fast pandas code runs when using vectorized operations compared to loops.

How does the time needed change as the data size grows?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

n = 10  # Define n before using it

df = pd.DataFrame({"value": range(n)})

# Vectorized operation
result_vec = df["value"] * 2

# Loop operation
result_loop = []
for v in df["value"]:
    result_loop.append(v * 2)

This code creates a DataFrame with n rows and doubles the values using two methods: vectorized and loop.

Identify Repeating Operations

• Primary operation: Multiplying each value by 2.
• How many times: Once per element, n times.
• Vectorized method does this internally in optimized C code without explicit Python loops.
• Loop method explicitly repeats the multiply operation n times in Python.

How Execution Grows With Input

As n grows, the number of multiply operations grows linearly.

Input Size (n)	Approx. Operations
10	10 multiplications
100	100 multiplications
1000	1000 multiplications

Pattern observation: The work doubles when the input doubles, growing in a straight line.

Final Time Complexity

Time Complexity: O(n)

This means the time needed grows directly in proportion to the number of rows.

Common Mistake

[X] Wrong: "Vectorized operations always run in constant time regardless of data size."

[OK] Correct: Vectorized operations still process each element, so time grows with data size, but they run faster because they avoid slow Python loops.

Interview Connect

Understanding how vectorized operations scale helps you write faster pandas code and shows you know how to handle data efficiently in real projects.

Self-Check

"What if we replaced the loop with a list comprehension? How would the time complexity change?"

Practice

1. Which of the following best describes vectorized operations in pandas?

easy

A. They require writing explicit for-loops to manipulate data.

B. They process data one row at a time using loops.

C. They operate on entire columns or arrays at once, making them faster.

D. They are slower than loops for large datasets.

Solution

1. Step 1: Understand vectorized operations

   Vectorized operations apply a function to whole columns or arrays simultaneously.

2. Step 2: Compare with loops

   Loops process data row by row, which is slower for large data.

3. Final Answer:

   They operate on entire columns or arrays at once, making them faster. -> Option C

4. Quick Check:

   Vectorized = whole columns at once [OK]

Hint: Vectorized means whole column at once, not row by row [OK]

Common Mistakes:

• Confusing vectorized operations with loops
• Thinking loops are faster for large data
• Believing vectorized operations require explicit loops

2. Which of the following is the correct syntax to add 10 to every element in a pandas Series s using vectorized operations?

easy

A. for i in s: s[i] = s[i] + 10

B. s = s + 10

C. s.apply(lambda x: x + 10)

D. s.add(10, inplace=True)

Solution

1. Step 1: Identify vectorized addition syntax

   Adding a scalar to a Series directly uses vectorized addition: s = s + 10.

2. Step 2: Check other options

   for i in s: s[i] = s[i] + 10 uses a loop incorrectly, C returns a new Series but does not assign, B is invalid because add does not have inplace.

3. Final Answer:

   s = s + 10 -> Option B

4. Quick Check:

   Use s + 10 for vectorized add [OK]

Hint: Use s + 10 for vectorized addition, no loops [OK]

Common Mistakes:

• Trying to modify Series inside a for-loop
• Using apply without assignment
• Using invalid inplace parameter with add()

3. What will be the output of this code?

import pandas as pd
s = pd.Series([1, 2, 3])
result = []
for val in s:
    result.append(val * 2)
print(result)

medium

A. Error: cannot multiply Series elements in loop

B. pd.Series([2, 4, 6])

C. [1, 2, 3]

D. [2, 4, 6]

Solution

1. Step 1: Understand the loop operation

   The loop goes through each value in the Series and multiplies it by 2, appending to a list.

2. Step 2: Check the printed output

   The printed output is the list [2, 4, 6], not a Series.

3. Final Answer:

   [2, 4, 6] -> Option D

4. Quick Check:

   Loop multiplies each value, list output [OK]

Hint: Loop appends multiplied values to list, not Series [OK]

Common Mistakes:

• Expecting a Series output instead of list
• Confusing loop with vectorized operation
• Thinking loop causes error multiplying values

4. Identify the error in this code that tries to double values in a pandas Series s:

result = []
for i in range(len(s)):
    result.append(s[i] * 2)
print(result)

medium

A. No error; code works correctly

B. Accessing Series elements by integer index is invalid

C. Using range(len(s)) causes an IndexError

D. Appending to list inside loop is not allowed

Solution

1. Step 1: Check indexing method

   Accessing Series elements by integer position with s[i] works if Series index is default 0-based.

2. Step 2: Verify loop and append

   The loop appends doubled values to list correctly; no error occurs.

3. Final Answer:

   No error; code works correctly -> Option A

4. Quick Check:

   Indexing by position works if default index [OK]

Hint: Default Series index allows integer access [OK]

Common Mistakes:

• Assuming s[i] always causes error
• Thinking append inside loop is invalid
• Confusing label-based and position-based indexing

5. You have a DataFrame df with a column 'price'. You want to increase all prices by 5% efficiently. Which method is best?

hard

A. Use df['price'] = df['price'] * 1.05 to apply vectorized multiplication.

B. Use a for-loop to multiply each price by 1.05 and update the DataFrame row by row.

C. Use df.apply(lambda row: row['price'] * 1.05, axis=1) without assignment.

D. Convert the column to a list, multiply each element by 1.05 in a loop, then assign back.

Solution

1. Step 1: Identify efficient vectorized method

   Multiplying the entire column by 1.05 directly uses vectorized operations and is fastest.

2. Step 2: Evaluate other options

   Use a for-loop to multiply each price by 1.05 and update the DataFrame row by row. uses slow loops, C does not assign result back, D is inefficient with list conversion.

3. Final Answer:

   Use df['price'] = df['price'] * 1.05 to apply vectorized multiplication. -> Option A

4. Quick Check:

   Vectorized column operation is fastest [OK]

Hint: Multiply whole column directly for speed [OK]

Common Mistakes:

• Using loops instead of vectorized operations
• Forgetting to assign result back to DataFrame
• Converting to list unnecessarily