diff --git a/py_scripts/digger.py b/dwarves/digger.py similarity index 82% rename from py_scripts/digger.py rename to dwarves/digger.py index 17b46fe..55d8fbe 100644 --- a/py_scripts/digger.py +++ b/dwarves/digger.py @@ -1,7 +1,7 @@ # Getting and combining data import pandas as pd import numpy as np -import scout +from dwarves import scout from numpy import ndarray from scipy.stats import binned_statistic @@ -40,13 +40,18 @@ def slam_dunk(dataset, column, labels): x_data = np.arange(0, len(dataset)) y_data = dataset[column] - x_bins, bin_edges, misc = binned_statistic(y_data, x_data, statistic='median', bins=len(labels)) + x_bins, bin_edges, misc = binned_statistic( + y_data, x_data, statistic="median", bins=len(labels) + ) bin_intervals = pd.IntervalIndex.from_arrays(bin_edges[:-1], bin_edges[1:]) - dataset['bin_value'] = dataset[column].apply(lambda x: set_to_median(x, bin_intervals)) + dataset["bin_value"] = dataset[column].apply( + lambda x: set_to_median(x, bin_intervals) + ) return dataset + def set_to_median(x, bin_intervals): for interval in bin_intervals: if x in interval: - return interval.mid \ No newline at end of file + return interval.mid diff --git a/py_scripts/engineer.py b/dwarves/engineer.py similarity index 100% rename from py_scripts/engineer.py rename to dwarves/engineer.py diff --git a/py_scripts/gunner.py b/dwarves/gunner.py similarity index 100% rename from py_scripts/gunner.py rename to dwarves/gunner.py diff --git a/py_scripts/mining_hq.py b/dwarves/mining_hq.py similarity index 100% rename from py_scripts/mining_hq.py rename to dwarves/mining_hq.py diff --git a/py_scripts/scout.py b/dwarves/scout.py similarity index 100% rename from py_scripts/scout.py rename to dwarves/scout.py