Correlation (correlate) in SciPy - Practice Problems & Coding Challenges

Challenge - 5 Problems

❓ Predict Output

intermediate

Output of 1D correlation using scipy.correlate

What is the output of the following code snippet using scipy.signal.correlate on two 1D arrays?

SciPy

import numpy as np
from scipy.signal import correlate

x = np.array([1, 2, 3])
y = np.array([0, 1, 0.5])
result = correlate(x, y, mode='full')
print(result)

A [0. 1. 2.5 4. 1.5]

B [0. 1. 2.5 3. 1.5]

C [0. 1. 2. 3. 1.5]

D [0. 1. 2.5 4. 3. ]

💡 Hint

Remember that correlation flips the second array before sliding it over the first.

The correlation flips y and slides it over x, computing sums of products at each shift. The output length is len(x)+len(y)-1 = 5. The values correspond to the sums at each shift.

❓ data_output

intermediate

Length of correlation output with different modes

Given two arrays a and b of lengths 4 and 3 respectively, what is the length of the output from scipy.signal.correlate(a, b, mode='valid')?

SciPy

import numpy as np
from scipy.signal import correlate

a = np.array([1, 2, 3, 4])
b = np.array([0, 1, 0.5])
result = correlate(a, b, mode='valid')
print(len(result))

A 3

B 2

C 4

D 6

💡 Hint

For 'valid' mode, output length is max(len(a), len(b)) - min(len(a), len(b)) + 1.

The 'valid' mode returns only points where the arrays fully overlap. So length is 4 - 3 + 1 = 2.

🔧 Debug

advanced

Identify the error in correlation code

What error will this code raise when run, and why?

SciPy

from scipy.signal import correlate

x = [1, 2, 3]
y = [0, 1, 'a']
result = correlate(x, y)
print(result)

A SyntaxError: invalid syntax

B ValueError: operands could not be broadcast together

C No error, outputs a numeric array

D TypeError: unsupported operand type(s) for *: 'int' and 'str'

💡 Hint

Check the data types in the input arrays.

The list y contains a string 'a', which cannot be multiplied by integers during correlation, causing a TypeError.

🚀 Application

advanced

Using correlation to find a pattern in a signal

You have a signal array signal = [1, 3, 2, 1, 0, 1, 3, 2] and a pattern pattern = [1, 3, 2]. Using scipy.signal.correlate with mode='valid', which index in the correlation output corresponds to the best match of the pattern in the signal?

SciPy

import numpy as np
from scipy.signal import correlate

signal = np.array([1, 3, 2, 1, 0, 1, 3, 2])
pattern = np.array([1, 3, 2])
correlation = correlate(signal, pattern, mode='valid')
best_match_index = np.argmax(correlation)
print(best_match_index)

A 5

B 0

C 1

D 6

💡 Hint

Look for where the pattern aligns best with the signal by highest correlation value.

The highest correlation value occurs at index 5, where the pattern matches the last three elements of the signal exactly.

🧠 Conceptual

expert

Effect of correlation mode on output length

Given two arrays of lengths 7 and 5, which mode of scipy.signal.correlate will produce an output array of length 5?

A 'full'

B 'valid'

C None of the above

D 'same'

💡 Hint

The 'same' mode returns output length equal to the first input array.

The 'same' mode returns an output with the same length as the first input array (7). 'valid' returns length 3, 'full' returns length 11. None produce length 5 except 'same' if the first array length is 5.

Practice

1. What does the scipy.signal.correlate function measure between two arrays?

easy

A. The difference between the maximum values of two arrays

B. The sum of elements in both arrays

C. The similarity between two arrays at different shifts

D. The product of the two arrays element-wise

Solution

1. Step 1: Understand correlation concept

   Correlation measures how similar two data sets are when one is shifted over the other.

2. Step 2: Identify function purpose

   scipy.signal.correlate computes this similarity at all possible shifts.

3. Final Answer:

   The similarity between two arrays at different shifts -> Option C

4. Quick Check:

   Correlation = similarity at shifts [OK]

Hint: Correlation = similarity at shifts between arrays [OK]

Common Mistakes:

• Confusing correlation with element-wise multiplication
• Thinking it sums elements instead of comparing shifts
• Mixing correlation with difference or subtraction

2. Which of the following is the correct way to import the correlate function from scipy.signal?

easy

A. import correlate from scipy.signal

B. from scipy.signal import correlate

C. from scipy import correlate

D. import scipy.correlate

Solution

1. Step 1: Recall correct import syntax

   Python imports functions using from module import function syntax.

2. Step 2: Match function and module

   correlate is in scipy.signal, so correct import is from scipy.signal import correlate.

3. Final Answer:

   from scipy.signal import correlate -> Option B

4. Quick Check:

   Correct import syntax = from scipy.signal import correlate [OK]

Hint: Use 'from scipy.signal import correlate' to import [OK]

Common Mistakes:

• Using 'import scipy.correlate' which is invalid
• Trying 'from scipy import correlate' missing submodule
• Incorrect order like 'import correlate from scipy.signal'

3. What is the output of this code?

import numpy as np
from scipy.signal import correlate
x = np.array([1, 2, 3])
y = np.array([0, 1, 0.5])
result = correlate(x, y, mode='full')
print(result)

medium

A. [0.5 2. 3.5 3. 0. ]

B. [0. 1. 2.5 3. 1.5]

C. [0. 1. 2. 3. 4.]

D. [1. 2. 3. 0. 0.5]

Solution

1. Step 1: Understand correlation calculation

   Correlation slides the reversed y over x and sums element-wise products at each shift. Equivalent to np.convolve(x, np.flip(y), mode='full').

2. Step 2: Calculate each position manually

   Reversed y = [0.5, 1, 0]
   Pos 0: 1*0.5 = 0.5
   Pos 1: 1*1 + 2*0.5 = 2.0
   Pos 2: 1*0 + 2*1 + 3*0.5 = 3.5
   Pos 3: 2*0 + 3*1 = 3.0
   Pos 4: 3*0 = 0.0
   result = [0.5 2. 3.5 3. 0. ]

3. Final Answer:

   [0.5 2. 3.5 3. 0. ] -> Option A

4. Quick Check:

   Cross-correlation with flip = [0.5 2. 3.5 3. 0.] [OK]

Hint: Calculate sums at each shift for full mode [OK]

Common Mistakes:

• Confusing convolution with correlation
• Miscounting output length for 'full' mode
• Ignoring zero padding in calculation

4. Identify the error in this code snippet:

import numpy as np
from scipy.signal import correlate
x = np.array([1, 2, 3])
y = np.array([4, 5])
result = correlate(x, y, mode='same')
print(result)

medium

A. The mode 'same' is invalid for correlate

B. Arrays x and y must be the same length

C. The import statement is incorrect

D. No error; code runs correctly and prints output

Solution

1. Step 1: Check mode validity

   The mode 'same' is valid and returns output same size as first input.

2. Step 2: Check array length requirements

   Arrays can have different lengths; correlate handles this with zero padding.

3. Step 3: Verify import correctness

   Import statement is correct as per scipy.signal documentation.

4. Final Answer:

   No error; code runs correctly and prints output -> Option D

5. Quick Check:

   Mode 'same' works with different length arrays [OK]

Hint: Mode 'same' works even if arrays differ in length [OK]

Common Mistakes:

• Thinking arrays must be equal length
• Assuming 'same' mode is invalid
• Misreading import statement as wrong

5. You have two time series arrays representing sensor readings:

sensor1 = [1, 3, 2, 4, 5]
sensor2 = [0, 1, 0.5, 2]

You want to find the shift where sensor2 best matches sensor1 using correlation. Which code snippet correctly finds the shift index with the highest correlation?

hard

A. import numpy as np
from scipy.signal import correlate
corr = correlate(sensor1, sensor2, mode='full')
shift = np.argmax(corr) - (len(sensor2) - 1)
print(shift)

B. import numpy as np
from scipy.signal import correlate
corr = correlate(sensor2, sensor1, mode='valid')
shift = np.argmax(corr)
print(shift)

C. import numpy as np
from scipy.signal import correlate
corr = correlate(sensor1, sensor2, mode='same')
shift = np.argmax(corr)
print(shift)

D. import numpy as np
from scipy.signal import correlate
corr = correlate(sensor1, sensor2)
shift = np.argmin(corr)
print(shift)

Solution

1. Step 1: Use 'full' mode for complete correlation

   To find best matching shift, full correlation output is needed to cover all shifts.

2. Step 2: Calculate shift index correctly

   Shift index = position of max correlation minus (length of second array - 1) to center shifts around zero.

3. Step 3: Verify code correctness

   import numpy as np from scipy.signal import correlate corr = correlate(sensor1, sensor2, mode='full') shift = np.argmax(corr) - (len(sensor2) - 1) print(shift) uses full mode, np.argmax, and correct shift calculation.

4. Final Answer:

   import numpy as np from scipy.signal import correlate corr = correlate(sensor1, sensor2, mode='full') shift = np.argmax(corr) - (len(sensor2) - 1) print(shift) -> Option A

5. Quick Check:

   Full mode + argmax - (len(y)-1) = correct shift [OK]

Hint: Use full mode and adjust argmax by (len(y)-1) for shift [OK]

Common Mistakes:

• Using 'valid' or 'same' mode which limits shifts
• Using argmin instead of argmax
• Not adjusting shift index by array length