Vectorized operations vs loops in Pandas

Introduction

Vectorized operations let you do many calculations at once, making your code faster and simpler. Loops do one step at a time, which can be slower and harder to read.

When you want to add or multiply all numbers in a column quickly.

When you need to apply the same calculation to every row in a table.

When you want to avoid writing long, slow loops over data.

When working with large datasets where speed matters.

When you want cleaner and easier-to-understand code.

Syntax

Pandas

result = df['column'] + 10  # Vectorized operation

for i in range(len(df)):
    df.loc[i, 'new_column'] = df.loc[i, 'column'] + 10  # Loop

Vectorized operations work on whole columns or arrays at once.

Loops go row by row, which is slower for big data.

Examples

This doubles every number in the 'value' column using vectorized operation.

Pandas

df['double'] = df['value'] * 2

This does the same doubling but with a loop, which is slower.

Pandas

for i in range(len(df)):
    df.loc[i, 'double'] = df.loc[i, 'value'] * 2

Vectorized check if age is 18 or more for all rows at once.

Pandas

df['is_adult'] = df['age'] >= 18

Sample Program

This code creates a small table, doubles the 'value' column using both vectorized operation and a loop, then prints the table and time taken by each method.

Pandas

import pandas as pd
import time

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

# Vectorized operation
start_vec = time.time()
df['double_vec'] = df['value'] * 2
end_vec = time.time()

# Loop operation
start_loop = time.time()
for i in range(len(df)):
    df.loc[i, 'double_loop'] = df.loc[i, 'value'] * 2
end_loop = time.time()

print('DataFrame after operations:')
print(df)
print(f"Vectorized time: {end_vec - start_vec:.6f} seconds")
print(f"Loop time: {end_loop - start_loop:.6f} seconds")

Important Notes

Vectorized operations are usually much faster than loops, especially on big data.

Loops can be easier to understand for very simple or custom logic but slow down with large data.

Use vectorized operations whenever possible for better performance and cleaner code.

Summary

Vectorized operations work on whole columns at once and are faster.

Loops process data row by row and are slower for large datasets.

Prefer vectorized operations in pandas for speed and simplicity.

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