# %%
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
import os

# note: we assume that you will execute from the directory of this code
# check your current directory
print("Current Working Directory:", os.getcwd())



# %%
data_path = "../../data_preprocess/exports/combined_group_allocation.csv"
df = pd.read_csv(data_path)

# %%
df
# %%
print('mean', df[df['Allocation'] == 'BGKF']['Comb_count'].mean())
print('std', df[df['Allocation'] == 'BGKF']['Comb_count'].std())
max = df[df['Allocation'] == 'BGKF']['Comb_count'].max()
min = df[df['Allocation'] == 'BGKF']['Comb_count'].min()
print('max', max)
print('max', min)
print('max - min', max - min)

# %%
print('mean', df[df['Allocation'] == 'GKF']['Comb_count'].mean())
print('std', df[df['Allocation'] == 'GKF']['Comb_count'].std())
max = df[df['Allocation'] == 'GKF']['Comb_count'].max()
min = df[df['Allocation'] == 'GKF']['Comb_count'].min()
print('max', max)
print('max', min)
print('max - min', max - min)


# %%