analysis.py

import pandas as pd
import matplotlib.pyplot as plt
import matplotlib

# dark mode toggle
darkmode = False

# theme colors
class Colors:
    alpha = "dd"
    GREEN = "#32a852" + alpha
    YELLOW = "#c9a132" + alpha
    RED = "#ad2626" + alpha
    BLUE = "#3399ff" + alpha
    BACKGROUND = (0.133, 0.153, 0.18) if darkmode else None

# creates a pie chart from an iterable
def agr_pie(lst, labels=None, colors=None, title=None):
    plt.figure().patch.set_facecolor(Colors.BACKGROUND)
    lst = list(lst)
    if labels == None:
        labels = list(set(lst))
    plt.pie(
        list( [lst.count(val) for val in labels] ),
        labels = list(
            [f"{label}\n{lst.count(label)} ({round(lst.count(label)/len(lst)*100, 1)}%)" for label in labels]
        ),
        colors = colors,
        labeldistance = 0.5,
    )
    plt.tight_layout()
    title_obj = plt.title(title, fontweight="bold")
    plt.setp(title_obj, color='gray')

# saves current fig to a given file
def saveplot(filename):
    plt.savefig(
        f'analysis_visualizations/{filename}',
        transparent=True,
        bbox_inches = 'tight',
        pad_inches=0
    )

# global formatting
matplotlib.rcParams['text.color'] = 'w'
matplotlib.rcParams['font.family'] = 'sans-serif'
matplotlib.rcParams['font.sans-serif'] = 'Segoe UI'
matplotlib.rcParams['font.size'] = '12'
matplotlib.rcParams['font.weight'] = 'bold'
matplotlib.rc('axes',edgecolor='gray')

# import data
df = pd.read_csv('analysis.csv')

# solved problems by difficulty
agr_pie(
    df[df["Solved?"]=="Yes"]["Difficulty"],
    labels=["Easy", "Medium", "Hard"],
    colors=[Colors.GREEN, Colors.YELLOW, Colors.RED],
    title="Solved problems by difficulty",
)
saveplot("pie_difficulty.png")

# does a more optimized solution exist?
agr_pie(
    df["Better optimization?"],
    labels=["Yes", "No"],
    colors=[Colors.RED, Colors.GREEN],
    title="Does a more optimized solution exist?",
)
saveplot("pie_optimization.png")

# problems solved out of attempted
agr_pie(
    ["Solved" if i < len(df[df["Solved?"]=="Yes"]) else "Failed" for i in range(len(df))],
    labels=["Solved", "Failed"],
    colors=[Colors.GREEN, Colors.RED],
    title="How many solved of those attempted?",
)
saveplot("pie_attempt.png")

# Time taken to solve problem by difficulty
plt.figure(figsize=(13, 4))
plt.hist(df[df["Difficulty"]=="Easy"]["Stopwatch"], bins=range(0,130,5), facecolor=Colors.GREEN, density=1, alpha=0.5)
plt.hist(df[df["Difficulty"]=="Medium"]["Stopwatch"], bins=range(0,130,5), facecolor=Colors.YELLOW, density=1, alpha=0.5)
plt.hist(df[df["Difficulty"]=="Hard"]["Stopwatch"], bins=range(0,120,5), facecolor=Colors.RED, density=1, alpha=0.5)
plt.tight_layout()
text_objs = []
text_objs.append(plt.title("Histogram of time taken to solve problem by difficulty", fontweight="bold"))
text_objs.append(plt.xlabel("Time (min)"))
text_objs.append(plt.ylabel("Density"))
for t in text_objs:
    plt.setp(t, color='gray')
plt.tick_params(axis='x', colors='gray')
plt.tick_params(axis='y', colors='gray')
saveplot("hist.png")