Feat(Fortune): Probability analyzer (#58)

scripts/plot_gen.py

@@ -0,0 +1,68 @@
+import json
+
+import matplotlib.pyplot as plt
+
+with open("./res.txt") as res_file:
+    res = json.load(res_file)
+
+good_len = -1
+bad_len = -1
+with open("../fortune_generator/json/fortune.json") as fortune_file:
+    j = json.load(fortune_file)
+    good_len = len(j["goodFortunes"])
+    bad_len = len(j["badFortunes"])
+
+status_bucket = [0] * 8
+good_bucket = [0] * good_len
+bad_bucket = [0] * bad_len
+
+for ip, v in res.items():
+    assert all(val != -1 for val in v)
+    good_bucket[v[0]] += 1
+    good_bucket[v[1]] += 1
+    bad_bucket[v[2]] += 1
+    bad_bucket[v[3]] += 1
+    status_bucket[v[4]] += 1
+
+
+groups = 1
+fig, axs = plt.subplots(groups, 1, figsize=(8, 6))
+
+axs.bar(
+    range(good_len),
+    good_bucket,
+    color="skyblue",
+    edgecolor="black",
+)
+axs.set_xlabel("Good Fortune Event Index")
+axs.set_ylabel("Occurrences")
+
+plt.tight_layout()
+plt.show()
+
+fig, axs = plt.subplots(groups, 1, figsize=(8, 6))
+
+axs.bar(
+    range(bad_len),
+    bad_bucket,
+    color="skyblue",
+    edgecolor="black",
+)
+axs.set_xlabel("Bad Fortune Event Index")
+axs.set_ylabel("Occurrences")
+
+plt.tight_layout()
+plt.show()
+
+fig, axs = plt.subplots(groups, 1, figsize=(8, 6))
+
+axs.bar(
+    range(len(status_bucket)),
+    status_bucket,
+    color="skyblue",
+    edgecolor="black",
+)
+axs.set_xlabel("Status Index")
+axs.set_ylabel("Occurrences")
+
+plt.show()