Skip to content

Commit 8ec8219

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent e6eb640 commit 8ec8219

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

maths/autocorrelation.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,17 +34,17 @@ def autocorrelation(data: list[float], lag: int) -> float:
3434
if variance == 0:
3535
raise ValueError("Variance of data is zero, autocorrelation undefined.")
3636

37-
covariance = sum(
38-
(data[i] - mean) * (data[i - lag] - mean)
39-
for i in range(lag, n)
40-
) / n
37+
covariance = (
38+
sum((data[i] - mean) * (data[i - lag] - mean) for i in range(lag, n)) / n
39+
)
4140

4241
return covariance / variance
4342

4443

4544
if __name__ == "__main__":
4645
import doctest
46+
4747
doctest.testmod()
4848
data = [1, 2, 3, 4, 5, 4, 3, 2, 1]
4949
for lag in range(5):
50-
print(f"Lag {lag}: {autocorrelation(data, lag):.4f}")
50+
print(f"Lag {lag}: {autocorrelation(data, lag):.4f}")

0 commit comments

Comments
 (0)