Correlation analysis (Pearson, Spearman) in Data Analysis Python - Time & Space Complexity

Time Complexity: Correlation analysis (Pearson, Spearman)

O(n log n)

Understanding Time Complexity

We want to know how the time needed to calculate correlation changes as the data grows.

How does the work increase when we have more data points?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd
from scipy.stats import pearsonr, spearmanr

def compute_correlations(df, col1, col2):
    pearson_corr, _ = pearsonr(df[col1], df[col2])
    spearman_corr, _ = spearmanr(df[col1], df[col2])
    return pearson_corr, spearman_corr

This code calculates Pearson and Spearman correlation between two columns in a data table.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Pearson: Single passes over data to compute means, std devs, covariance: O(n).
• Spearman: Sorting to compute ranks: O(n log n), followed by linear correlation on ranks: O(n).
• Dominant operation: Sorting in Spearman.

How Execution Grows With Input

As the number of data points n increases, the time grows as O(n log n), dominated by sorting for Spearman ranks.

Input Size (n)	Approx. Operations
10	About 100 (linear passes + sorting ~2×n log n)
100	About 1,500 (linear + sorting)
1000	About 20,000 (linear + sorting)

Pattern observation: Grows faster than linear due to log n factor in sorting, but still very efficient.

Final Time Complexity

Time Complexity: O(n log n)

This is dominated by the sorting step required for Spearman ranking; Pearson is O(n).

Common Mistake

[X] Wrong: "Correlation calculation time grows much faster than the data size because it compares every pair of points."

[OK] Correct: No pairwise comparisons (that would be O(n²)). Pearson uses O(n) summary statistics; Spearman uses O(n log n) sorting for ranks.

Interview Connect

Understanding how correlation calculations scale helps you explain data analysis efficiency clearly and confidently.

Self-Check

"What if we wanted to compute correlations between every pair of columns in a dataset? How would the time complexity change?"

Practice

1. What does the Pearson correlation coefficient measure between two variables?

easy

A. The rank order similarity between two variables

B. The difference in means between two groups

C. The strength and direction of a linear relationship

D. The frequency distribution of a variable

Solution

1. Step 1: Understand Pearson correlation concept

   Pearson correlation measures how strongly two variables move together in a straight line.

2. Step 2: Differentiate from other measures

   Unlike Spearman, Pearson focuses on linear relationships, not ranks or means.

3. Final Answer:

   The strength and direction of a linear relationship -> Option C

4. Quick Check:

   Pearson = linear relationship [OK]

Hint: Pearson = linear, Spearman = rank-based [OK]

Common Mistakes:

• Confusing Pearson with Spearman correlation
• Thinking Pearson measures differences in means
• Assuming Pearson works for non-linear relationships

2. Which of the following Python code snippets correctly calculates the Spearman correlation between two pandas Series x and y?

easy

A. x.corr(y, method='spearman')

B. x.corr(y, method='pearson')

C. spearmanr(x, y)

D. corr(x, y, method='spearman')

Solution

1. Step 1: Recall pandas correlation syntax

   In pandas, Series.corr() accepts a method argument with values like 'pearson' or 'spearman'.

2. Step 2: Identify correct method call

   x.corr(y, method='spearman') uses x.corr(y, method='spearman'), which is the correct pandas syntax.

3. Final Answer:

   x.corr(y, method='spearman') -> Option A

4. Quick Check:

   pandas Series.corr with method='spearman' [OK]

Hint: Use Series.corr with method='spearman' for Spearman correlation [OK]

Common Mistakes:

• Using 'pearson' instead of 'spearman' for Spearman correlation
• Calling a non-existent function like corr() standalone
• Confusing scipy's spearmanr with pandas syntax

3. Given the following code, what is the output of print(round(corr, 2))?

import pandas as pd
x = pd.Series([1, 2, 3, 4, 5])
y = pd.Series([5, 6, 7, 8, 8])
corr = x.corr(y, method='pearson')

medium

A. 0.75

B. 0.90

C. 0.85

D. 0.97

Solution

1. Step 1: Calculate Pearson correlation manually

   Calculate covariance and standard deviations of x and y, then compute correlation. Here, x and y increase mostly together, so correlation is high.

2. Step 2: Use pandas to confirm

   Running the code gives approximately 0.9746, which rounds to 0.97.

3. Final Answer:

   0.97 -> Option D

4. Quick Check:

   pandas Pearson correlation ≈ 0.97 [OK]

Hint: Use pandas corr() to get exact correlation quickly [OK]

Common Mistakes:

• Confusing Pearson with Spearman correlation values
• Rounding incorrectly or too early
• Assuming correlation must be exactly 1 for similar sequences

4. Identify the error in this code snippet that attempts to calculate Spearman correlation:

from scipy.stats import spearmanr
x = [1, 2, 3, 4]
y = [4, 3, 2, 1]
result = spearmanr(x, y)
print(result.correlation)

medium

A. The function spearmanr requires numpy arrays, not lists

B. The attribute 'correlation' does not exist in the result object

C. The print statement should be inside a function

D. The import statement is incorrect

Solution

1. Step 1: Check spearmanr return type

   spearmanr returns a namedtuple SpearmanRResult with fields statistic (Spearman correlation coefficient) and pvalue.

2. Step 2: Confirm attribute access

   The code accesses result.correlation, but no such attribute exists. Use result.statistic instead.

3. Final Answer:

   The attribute 'correlation' does not exist in the result object -> Option B

4. Quick Check:

   spearmanr uses result.statistic [OK]

Hint: Use result.statistic for spearmanr correlation [OK]

Common Mistakes:

• Assuming lists cannot be used with spearmanr
• Misunderstanding the return type of spearmanr
• Incorrect import or syntax errors

5. You have two variables: age (numeric) and education_level (ordinal ranks: 1=High School, 2=Bachelor, 3=Master, 4=PhD). Which correlation method is most appropriate to measure their association and why?

hard

A. Spearman, because education_level is ordinal and relationship may be monotonic

B. Pearson, because it handles ordinal data better

C. Neither, because correlation cannot be calculated between these variables

D. Pearson, because both are numeric variables

Solution

1. Step 1: Identify variable types

   Age is numeric continuous; education_level is ordinal categorical (ranked).

2. Step 2: Choose correlation method

   Spearman correlation is best for ordinal data and monotonic relationships, unlike Pearson which assumes linear and interval data.

3. Final Answer:

   Spearman, because education_level is ordinal and relationship may be monotonic -> Option A

4. Quick Check:

   Ordinal data + monotonic = Spearman [OK]

Hint: Use Spearman for ordinal or ranked data [OK]

Common Mistakes:

• Using Pearson for ordinal data incorrectly
• Thinking correlation can't be done with ordinal variables
• Confusing monotonic with linear relationships