Feature transformers (VectorAssembler, StandardScaler) in Apache Spark - Time & Space Complexity

Time Complexity: Feature transformers (VectorAssembler, StandardScaler)

O(n)

Understanding Time Complexity

We want to understand how the time needed to transform data grows as the data size increases.

How does using VectorAssembler and StandardScaler affect processing time when data gets bigger?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

from pyspark.ml.feature import VectorAssembler, StandardScaler

assembler = VectorAssembler(inputCols=["feature1", "feature2"], outputCol="features")
assembled_data = assembler.transform(df)

scaler = StandardScaler(inputCol="features", outputCol="scaledFeatures", withMean=True, withStd=True)
scaler_model = scaler.fit(assembled_data)
scaled_data = scaler_model.transform(assembled_data)

This code combines columns into a vector and then scales the features to have mean zero and unit variance.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Processing each row of the dataset to assemble and scale features.
• How many times: Once per row for both assembling and scaling steps.

How Execution Grows With Input

As the number of rows grows, the time to process grows roughly in direct proportion.

Input Size (n)	Approx. Operations
10	About 10 times the work to assemble and scale features
100	About 100 times the work
1000	About 1000 times the work

Pattern observation: Doubling the data roughly doubles the work needed.

Final Time Complexity

Time Complexity: O(n)

This means the time grows linearly with the number of rows in the dataset.

Common Mistake

[X] Wrong: "Scaling features takes constant time no matter how big the data is."

[OK] Correct: Each row must be processed to compute and apply scaling, so time grows with data size.

Interview Connect

Knowing how feature transformers scale helps you explain data preparation steps clearly and shows you understand performance impacts in real projects.

Self-Check

"What if we added more features to assemble and scale? How would the time complexity change?"

Practice

1. What is the main purpose of the VectorAssembler in Apache Spark's feature transformers?

easy

A. To remove missing values from the dataset

B. To scale features to have zero mean and unit variance

C. To combine multiple columns into a single vector column

D. To split a vector column into multiple columns

Solution

1. Step 1: Understand VectorAssembler's role

   VectorAssembler is used to combine several feature columns into one vector column, which is needed for ML algorithms.

2. Step 2: Differentiate from other transformers

   StandardScaler scales features, but VectorAssembler only combines columns into a vector.

3. Final Answer:

   To combine multiple columns into a single vector column -> Option C

4. Quick Check:

   VectorAssembler = combine columns [OK]

Hint: VectorAssembler joins columns into one vector column [OK]

Common Mistakes:

• Confusing VectorAssembler with StandardScaler
• Thinking it splits vectors instead of combining
• Assuming it handles missing values

2. Which of the following is the correct way to create a VectorAssembler in PySpark to combine columns 'age' and 'income' into 'features'?

easy

A. VectorAssembler(inputCols=['age', 'income'], outputCol='features')

B. VectorAssembler(outputCols=['age', 'income'], inputCol='features')

C. VectorAssembler(inputCol=['age', 'income'], outputCols='features')

D. VectorAssembler(columns=['age', 'income'], output='features')

Solution

1. Step 1: Recall VectorAssembler syntax

   The correct parameters are inputCols (list of input columns) and outputCol (name of output vector column).

2. Step 2: Match parameters with options

   VectorAssembler(inputCols=['age', 'income'], outputCol='features') uses inputCols and outputCol correctly; others misuse parameter names or types.

3. Final Answer:

   VectorAssembler(inputCols=['age', 'income'], outputCol='features') -> Option A

4. Quick Check:

   inputCols and outputCol are correct parameters [OK]

Hint: Use inputCols and outputCol parameters exactly [OK]

Common Mistakes:

• Swapping inputCols and outputCol names
• Using singular inputCol instead of inputCols
• Using wrong parameter names like columns or output

3. Given the following PySpark code, what will be the output of scaled_features.show()?

from pyspark.ml.feature import VectorAssembler, StandardScaler
from pyspark.sql import SparkSession

spark = SparkSession.builder.getOrCreate()
data = [(1, 10.0, 100.0), (2, 20.0, 200.0), (3, 30.0, 300.0)]
df = spark.createDataFrame(data, ['id', 'feature1', 'feature2'])

assembler = VectorAssembler(inputCols=['feature1', 'feature2'], outputCol='features')
assembled = assembler.transform(df)

scaler = StandardScaler(inputCol='features', outputCol='scaledFeatures', withMean=True, withStd=True)
scalerModel = scaler.fit(assembled)
scaled_features = scalerModel.transform(assembled).select('id', 'scaledFeatures')

medium

A. A DataFrame with 'id' and 'scaledFeatures' columns where features are scaled to zero mean and unit variance

B. A DataFrame with 'id' and 'features' columns unchanged

C. An error because StandardScaler cannot be applied after VectorAssembler

D. A DataFrame with only 'scaledFeatures' column without 'id'

Solution

1. Step 1: Understand VectorAssembler output

   VectorAssembler creates a 'features' vector column combining 'feature1' and 'feature2'.

2. Step 2: Apply StandardScaler

   StandardScaler scales 'features' to zero mean and unit variance, outputting 'scaledFeatures'.

3. Final Answer:

   A DataFrame with 'id' and 'scaledFeatures' columns where features are scaled to zero mean and unit variance -> Option A

4. Quick Check:

   VectorAssembler + StandardScaler = scaled features [OK]

Hint: StandardScaler scales VectorAssembler output vector [OK]

Common Mistakes:

• Expecting original columns unchanged after scaling
• Thinking StandardScaler causes error after VectorAssembler
• Missing 'id' column in output

4. Identify the error in the following PySpark code snippet using StandardScaler:

assembler = VectorAssembler(inputCols=['f1', 'f2'], outputCol='features')
df2 = assembler.transform(df)
scaler = StandardScaler(inputCol='features', outputCol='scaled', withMean=False, withStd=True)
scaled_df = scaler.transform(df2)

medium

A. Wrong inputCol name in StandardScaler

B. Missing call to fit() before transform() on StandardScaler

C. VectorAssembler outputCol should be 'scaled' not 'features'

D. withMean must be True for StandardScaler to work

Solution

1. Step 1: Check StandardScaler usage

   StandardScaler requires calling fit() on data before transform().

2. Step 2: Identify missing fit() call

   The code calls transform() directly on scaler without fit(), causing error.

3. Final Answer:

   Missing call to fit() before transform() on StandardScaler -> Option B

4. Quick Check:

   StandardScaler needs fit() before transform() [OK]

Hint: Always fit StandardScaler before transform [OK]

Common Mistakes:

• Calling transform() without fit() on StandardScaler
• Confusing inputCol and outputCol names
• Assuming withMean must be True

5. You have a dataset with columns 'height', 'weight', and 'age'. You want to prepare features for a machine learning model by combining these columns into a single vector and scaling them to zero mean and unit variance. Which sequence of steps correctly achieves this in PySpark?

hard

A. Use StandardScaler first on raw columns, then combine with VectorAssembler

B. Apply StandardScaler separately on each column, then combine with VectorAssembler

C. Combine columns with VectorAssembler and use StandardScaler without fitting

D. Use VectorAssembler to combine columns, then fit and transform with StandardScaler on the assembled vector

Solution

1. Step 1: Combine columns using VectorAssembler

   VectorAssembler merges 'height', 'weight', and 'age' into one vector column for ML models.

2. Step 2: Fit and transform the assembled vector with StandardScaler

   StandardScaler requires fitting on the vector column to scale features properly.

3. Final Answer:

   Use VectorAssembler to combine columns, then fit and transform with StandardScaler on the assembled vector -> Option D

4. Quick Check:

   Combine first, then scale with fit() [OK]

Hint: Combine columns first, then fit StandardScaler [OK]

Common Mistakes:

• Scaling columns before combining
• Skipping fit() on StandardScaler
• Combining after scaling