From f158c5050beae82dcdc0ccdbe5e4c66597b277a2 Mon Sep 17 00:00:00 2001
From: Alexandr Velikiy <39257464+VargBurz@users.noreply.github.com>
Date: Tue, 18 Sep 2018 12:12:24 +0300
Subject: [PATCH] aggr model (#161)

* aggr model

new defs for aggregate model

* Update utils

raise exception
---
 analytics/utils/__init__.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/analytics/utils/__init__.py b/analytics/utils/__init__.py
index 89e6ab5..bb20e6c 100644
--- a/analytics/utils/__init__.py
+++ b/analytics/utils/__init__.py
@@ -206,3 +206,18 @@ def peak_finder(data, size):
         if data[i] == max(data[i - size: i + size]) and data[i] > data[i + 1]:
             all_max.append(i)
     return all_max
+
+def ar_mean(numbers):
+    return float(sum(numbers)) / max(len(numbers), 1)
+
+def get_av_model(patterns_list):
+    x = len(patterns_list[0])
+    if len(patterns_list[1]) != x:
+        raise NameError('All elements of patterns_list should have same length')
+    model_pat = []
+    for i in range(x):
+        av_val = []
+        for j in patterns_list:
+            av_val.append(j.values[i])
+        model_pat.append(ar_mean(av_val))
+    return model_pat