docs: 📝 update

Author: ivaquero

binfo.ipynb

@@ -1,2 +1,4 @@
 from stats_utils import confidence_ellipse
-from stats_utils import fetch_data
+from stats_utils import fetch_data
+from stats_utils.rnd_matrix import rnd_replace_with_const
+from stats_utils.record_rnd import record_by_time

stats_utils/__init__.py

@@ -0,0 +1,12 @@
+import itertools
+import pandas as pd
+
+
+def assign_matrix(*cols):
+    '''
+    Assigns a 0-1 matrix for each cols
+    '''
+    list_0_1 = [*product([0, 1], repeat=len(cols))]
+    assign_df = pd.DataFrame(list_0_1, columns=cols)
+    
+    return assign_df 

stats_utils/assign_matrix.py

@@ -0,0 +1,13 @@
+import numpy as np
+from datetime import datetime
+import pathlib
+
+def record_by_time(func_name, first_arg, output):
+    hour = datetime.now().hour
+    minute = datetime.now().minute
+
+    record_path = f'./record/{func_name}'
+
+    pathlib.Path(f'{record_path}').mkdir(parents=True, exist_ok=True)
+    np.savetxt(f'{record_path}/{first_arg}_{hour}_{minute}.txt', X=output)
+ 

stats_utils/record_rnd.py

@@ -0,0 +1,18 @@
+import numpy as np
+from skfake.record_rnd import *
+
+
+def rnd_replace_with_const(n_groups, n_samples, constant=0, col_ind=0, sort=-1, seed=None, record=None):
+    if seed:
+        np.random.seed(seed)
+    rnd_matrix = np.random.rand(n_groups, n_samples)
+    if sort:
+        rnd_matrix.sort(axis=sort)
+        
+    the_col = np.ones(n_groups) * constant
+    rnd_matrix[:, col_ind] = the_col
+
+    if record == 'time':   
+        record_by_time(rnd_2d_from.__name__, n_groups, output)
+    
+    return rnd_matrix