scipy.signal.correlation_lags

scipy.signal.correlation_lags(in1_len, in2_len, mode='full')[source]

Calculates the lag / displacement indices array for 1D cross-correlation.

Parameters
in1_sizeint

First input size.

in2_sizeint

Second input size.

modestr {‘full’, ‘valid’, ‘same’}, optional

A string indicating the size of the output. See the documentation correlate for more information.

Returns
lagsarray

Returns an array containing cross-correlation lag/displacement indices. Indices can be indexed with the np.argmax of the correlation to return the lag/displacement.

See also

correlate

Compute the N-dimensional cross-correlation.

Notes

Cross-correlation for continuous functions \(f\) and \(g\) is defined as:

\[\left ( f\star g \right )\left ( \tau \right ) \triangleq \int_{t_0}^{t_0 +T} \overline{f\left ( t \right )}g\left ( t+\tau \right )dt\]

Where \(\tau\) is defined as the displacement, also known as the lag.

Cross correlation for discrete functions \(f\) and \(g\) is defined as:

\[\left ( f\star g \right )\left [ n \right ] \triangleq \sum_{-\infty}^{\infty} \overline{f\left [ m \right ]}g\left [ m+n \right ]\]

Where \(n\) is the lag.

Examples

Cross-correlation of a signal with its time-delayed self.

>>> from scipy import signal
>>> from numpy.random import default_rng
>>> rng = default_rng()
>>> x = rng.standard_normal(1000)
>>> y = np.concatenate([rng.standard_normal(100), x])
>>> correlation = signal.correlate(x, y, mode="full")
>>> lags = signal.correlation_lags(x.size, y.size, mode="full")
>>> lag = lags[np.argmax(correlation)]