M1-Survey-Engine/tools/statistics/analyse/duration_per_experience.py
2024-01-24 16:14:15 +01:00

37 lines
1.1 KiB
Python

import matplotlib.pyplot as plt
import numpy as np
from tools.statistics import extract
def analyse(datas: list[dict]):
experience_duration: dict[str, int] = {"yes": 0, "mixed": 0, "no": 0}
experience_count: dict[str, int] = {"yes": 0, "mixed": 0, "no": 0}
for data in datas:
experience = extract.experience.extract(data)
experience_count[experience] += 1
for survey, survey_data in data["surveys"].items():
# only scan survey mission
if not survey.startswith("mission-"):
continue
if extract.mission_completed.extract(data, survey):
experience_duration[experience] += extract.mission_duration.extract(data, survey)
x = list(experience_duration.keys())
y = (
np.array(list(experience_duration.values()))
/ np.array(list(experience_count.values()))
)
# prepare plotting
figure: plt.Figure = plt.figure()
axes = figure.add_subplot(1, 1, 1)
axes.set_title("Temps moyen passé par expérience")
# bar chart
axes.bar(x, y)
plt.show(block=True)