Skip to content

Commit bd1e8e0

Browse files
committed
replace correlation_lags with calculation + new example
1 parent d3e7387 commit bd1e8e0

2 files changed

Lines changed: 271 additions & 1 deletion

File tree

control/stochsys.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -565,6 +565,8 @@ def correlation(T, X, Y=None, squeeze=True):
565565
[[sp.signal.correlate(X[i], Y[j])
566566
for i in range(X.shape[0])] for j in range(Y.shape[0])]
567567
) * dt / (T[-1] - T[0])
568-
tau = sp.signal.correlation_lags(len(X[0]), len(Y[0])) * dt
568+
# From scipy.signal.correlation_lags (for use with older versions)
569+
# tau = sp.signal.correlation_lags(len(X[0]), len(Y[0])) * dt
570+
tau = np.arange(-len(Y[0]) + 1, len(X[0])) * dt
569571

570572
return tau, R.squeeze() if squeeze else R

0 commit comments

Comments
 (0)