What are some examples of vectorization in Pandas?

What are some examples of vectorization in Pandas?

Vectorization in Pandas refers to the process of applying operations to entire arrays instead of using loops. This can significantly speed up computations. For example, you can add a constant to a DataFrame column directly.

Can you show me a code example?

Sure! Here's a simple example:

import pandas as pd

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

df['C'] = df['A'] + 10  # Vectorized operation
print(df)

This will add 10 to each element in column 'A' and store the result in column 'C'.

What about using vectorization for conditional operations?

You can use np.where for conditional vectorization. For example:

import numpy as np

df['D'] = np.where(df['A'] > 1, 'Greater', 'Lesser')

This will create a new column 'D' based on the condition applied to column 'A'.